Использование izip над вторым измерением Numpy массива

Question

я следующее:

for i in xrange(n): 
    label = labels[i] 
    frame = data[:, i] 
    dostuff() 

где данные 2-D numpy массив. Я хотел бы переписать его, используя izip.

for label, frame in izip(labels, ???): 
    dostuff() 

Что я могу заменить ???, чтобы заставить это работать?

Answer 1

Если вы действительно хотите использовать izip(), вы можете сделать это следующим образом:

for label, frame in izip(labels, data.T): 
    dostuff() 

В общем, желательно, чтобы не перебрать массив NumPy с помощью цикла Python, а использовать NumPy ufuncs делать петли в коде C. Как это сделать, зависит от того, что на самом деле делает dostuff().

Answer 2

Вы могли бы использовать:

??? = numpy.transpose(data) 

EDIT: удалить вторую версию. Поскольку транспозиция не копирует массив, как я полагал, нет никаких оснований для чего-то такого сумасшедшего, как я делал.