如何迭代矩阵的列？

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:

这可能看起来很昂贵，但实际上非常便宜（它将视图返回到相同的数据上，但其形状和步幅属性被置换）。

Answer 2

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

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