如何迭代矩阵的列？

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: 
    ...