python - 如何迭代矩阵的列？

Question

如果在 python 中定义:

a = arange(9).reshape(3,3)

作为 3x3 矩阵并进行迭代:

for i in a:

它将遍历矩阵的行。有什么方法可以遍历列吗？

Answer 1

怎么样

for i in a.transpose():

或者，更短:

for i in a.T:

这可能看起来很昂贵，但实际上非常便宜(它返回相同数据的 View ，但形状和步幅属性已置换)。