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

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