Он будет перебирать строки матрицы. Есть ли способ перебрать столбцы?
Ответ 1
Как насчет
for i in a.transpose():
или, короче:
for i in a.T:
Это может выглядеть дорого, но на самом деле очень дешево (он возвращает представление на одни и те же данные, но с измененными атрибутами формы и шага).
Ответ 2
Предполагая, что a является хорошо сформированной матрицей, вы можете попробовать что-то вроде: