如何遍历矩阵的列？

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:

这看起来可能很复杂，但实际上非常简单（它只是返回了同一数据的一个视图，只是形状和步幅属性进行了调整）。