How do you extract a column from a multi-dimensional array?

Question

Does anybody know how to extract a column from a multi-dimensional array in Python?

Answer 1

If you want to grab more than just one column just use slice:

 a = np.array([[1, 2, 3],[4, 5, 6],[7, 8, 9]])
    print(a[:, [1, 2]])
[[2 3]
[5 6]
[8 9]]