如何迭代矩陣的列？

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