how to get the Nth column of array of arrays in python

Question

I have array names y

and y.shape gives

y.shape
Out[6]: (9976, 158)

the value of y is

y
Out[3]: 
array([[0, 0, 1, ..., 0, 0, 0],
       [0, 0, 1, ..., 0, 0, 0],
       [0, 0, 0, ..., 0, 0, 0],
       ...,
       [0, 0, 0, ..., 0, 0, 0],
       [0, 0, 0, ..., 0, 0, 0],
       [1, 0, 0, ..., 0, 0, 0]])

I want to get the Nth column of each array

I tried y[N] but that gave the Nth row

so

y[0]
Out[4]: 
array([0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0,
       0, 0, 0, 0, 0, 0, 0, 1, 1, 0, 1, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0,
       0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
       0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
       0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
       0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
       0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 0, 1, 0,
       0, 0, 0, 0])

how can I do that?

Are you using numpy?

drum
– drum

2021-10-15 02:44:53 +00:00
Commented Oct 15, 2021 at 2:44 — drum
– drum, Commented Oct 15, 2021 at 2:44
i can use any libraru

asmgx
– asmgx

2021-10-15 02:47:24 +00:00
Commented Oct 15, 2021 at 2:47 — asmgx
– asmgx, Commented Oct 15, 2021 at 2:47
Numpy nth column of 2d array: y[:, n]

DarrylG
– DarrylG

2021-10-15 03:00:02 +00:00
Commented Oct 15, 2021 at 3:00 — DarrylG
– DarrylG, Commented Oct 15, 2021 at 3:00

mozway · Accepted Answer · 2021-10-15 02:59:46Z

4

Use 2D slicing: y[:, n]

: -> all values in the first dimension (rows)

n -> just the nth column

answered Oct 15, 2021 at 2:59

mozway

267k13 gold badges56 silver badges106 bronze badges

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

Collectives™ on Stack Overflow

how to get the Nth column of array of arrays in python

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related