Как перебирать столбцы матрицы?

В python, если define:

a = arange(9).reshape(3,3)

в качестве матрицы 3x3 и итерации:

for i in a:

Он будет перебирать строки матрицы. Есть ли способ перебрать столбцы?

Ответы

Ответ 1

Как насчет

for i in a.transpose():

или, короче:

for i in a.T:

Это может выглядеть дорого, но на самом деле очень дешево (он возвращает представление на одни и те же данные, но с измененными атрибутами формы и шага).

Ответ 2

Предполагая, что a является хорошо сформированной матрицей, вы можете попробовать что-то вроде:

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