如何从多维数组中提取列？

Question

有没有人知道怎么从一个多维数组中提取一列数据，在Python里怎么做呢？

Answer 1

如果你有一个数组，比如说：

a = [[1, 2], [2, 3], [3, 4]]

那么你可以这样提取第一列：

[row[0] for row in a]

这样得到的结果看起来是这样的：

[1, 2, 3]

Answer 2

你是不是在用 NumPy 数组？Python 有一个 数组 模块，但它不支持多维数组。普通的 Python 列表也是一维的。

不过，如果你有一个简单的二维列表，比如这样：

A = [[1,2,3,4],
     [5,6,7,8]]

那么你可以这样提取一列：

def column(matrix, i):
    return [row[i] for row in matrix]

提取第二列（索引为 1）：

>>> column(A, 1)
[2, 6]

或者你也可以简单地这样做：

>>> [row[1] for row in A]
[2, 6]

Answer 3

>>> import numpy as np
>>> A = np.array([[1,2,3,4],[5,6,7,8]])

>>> A
array([[1, 2, 3, 4],
    [5, 6, 7, 8]])

>>> A[:,2] # returns the third columm
array([3, 7])

nrows = 3
ncols = 4
my_array = numpy.arange(nrows*ncols, dtype='double')
my_array = my_array.reshape(nrows, ncols)

另请参见：“numpy.arange”和“reshape”来分配内存

示例：（分配一个形状为矩阵（3x4）的数组）