Suppose I have:

```
test = numpy.array([[1, 2], [3, 4], [5, 6]])
```

`test[i]`

gets me *ith* line of the array (eg `[1, 2]`

). How can I access the *ith* column? (eg `[1, 3, 5]`

). Also, would this be an expensive operation?

```
>>> test[:,0]
array([1, 3, 5])
```

Similarly,

```
>>> test[1,:]
array([3, 4])
```

lets you access rows. This is covered in Section 1.4 (Indexing) of the NumPy reference. This is quick, at least in my experience. It's certainly much quicker than accessing each element in a loop.

And if you want to access more than one column at a time you could do:

```
>>> test = np.arange(9).reshape((3,3))
>>> test
array([[0, 1, 2],
[3, 4, 5],
[6, 7, 8]])
>>> test[:,[0,2]]
array([[0, 2],
[3, 5],
[6, 8]])
```

Licensed under: CC-BY-SA with attribution

Not affiliated with: Stack Overflow