如何在矩阵的列上迭代？

Question

在python中，如果定义：

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

作为3x3矩阵并迭代：

for i in a:

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

Answer 1

假设a是一个格式良好的矩阵，您可以尝试如下操作：

b = zip(*a)
for index in b:
   ...

Answer 2

怎么样

for i in a.transpose():

或者，更短：

for i in a.T:

这可能看起来很昂贵，但实际上非常便宜（它将视图返回到相同的数据上，但形状和跨距属性是排列的）。