Splitting columns of a numpy array easily

Question

How can I split an array's columns into three arrays x, y, z without manually writing each of the [:,0],[:,1],[:,2] separately?

Example

# Create example np array
import numpy as np
data = np.array([[1,2,3],[4,5,6],[7,8,9]])

Now data is

[[1 2 3]
 [4 5 6]
 [7 8 9]]

What I want to do:

x, y, z = data[:,0], data[:,1], data[:,2] ## Help me here!
print(x)

Wanted output:

array([1, 4, 7])

behzad.nouri · Accepted Answer · 2015-06-13 16:32:02Z

32

Transpose, then unpack:

>>> x, y, z = data.T
>>> x
array([1, 4, 7])

answered Jun 13, 2015 at 16:32

behzad.nouri

78.5k18 gold badges130 silver badges127 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

Tanveer Alam · Accepted Answer · 2015-06-13 16:32:26Z

5

You don't need to slice it.

>>> import numpy as np
>>> data = np.array([[1,2,3],[4,5,6],[7,8,9]])
>>> x, y, z = data.T
>>> x
array([1, 4, 7])
>>> y
array([2, 5, 8])
>>> z
array([3, 6, 9])

answered Jun 13, 2015 at 16:32

Tanveer Alam

5,2754 gold badges24 silver badges44 bronze badges

Collectives™ on Stack Overflow

Splitting columns of a numpy array easily

Example

2 Answers 2

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

Example

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related