访问numpy.ndarray的列和行

Question

我目前正在努力从存储为numpy.ndarray的矩阵中提取某些列和行。

我有一个列表，其中我附加了这些numpy.ndarrays。

此列表存储在名为data

的变量中

print data[0].shape

输出此

(400, 288)

我根据文档已经了解到矩阵有400行和288列。

如何单独提取所有288个？

示例：

>> import numpy as np
>> data = np.random.rand(3,3)
>> print data

[[ 0.97522481  0.57583658  0.68582806]
 [ 0.88509883  0.22261933  0.84307038]
 [ 0.59397925  0.51592125  0.54346909]]

如何单独打印此3x3矩阵的列，首先是

[0.97522481 , 0.88509883, 0.59397925 ]

没有输出其他人？

Answer 1

这是你在找什么？

import numpy as np
arr = np.array([[1, 2], 
                [3, 4], 
                [5, 6]])
print(arr.shape)
# (3, 2)
print(list(data.T))
# [array([1, 3, 5]), array([2, 4, 6])]