python - 如何应用切片运算符来获取python中数组的第n列？

Question

我有 size 的二维矩阵n，我想将整个n-1th 列值放入另一个列表中。例如，

a = [[1, 2], [3, 4], [5, 6]]
a[:][0] // return [1,2]

如何a使用切片运算符将上述二维数组的 1,3,5 放入列表中

标签： pythonarrayspython-3.x

Answer 1

您可以使用numpy库执行此操作：

import numpy

a = np.array([[1, 2], [3, 4], [5, 6]])
result = a[:, 0]   # Returns a 1-D numpy array [1, 3, 5]

可以在这里找到更高级的索引和切片选项。