如何迭代矩陣的列？

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