Переписать со списками: как ссылаться на первый элемент в списке

Question

Я хочу скопировать содержимое нескольких полей в массиве записей в ndarray (оба типа float64). Я знаю, как это сделать, когда данные повторного массива имеют одно значение в каждом поле:

my_ndarray[:,0]=my_recarray['X']  #(for field 'X')

Теперь у меня есть повторный массив со списком из 5 чисел с плавающей запятой в каждом поле, и я хочу скопировать только первый элемент каждого списка. Когда я использую вышеупомянутое с новым recarray (и списком), я получаю эту ошибку:

ValueError: could not broadcast input array from shape (92,5) into shape (92)

Это имеет полный смысл (задним числом).

Я думал, что смогу получить только первый элемент каждого с этим:

my_ndarray[:,0]=my_recarray['X'][0]  #(for field 'X')

Я получаю эту ошибку:

ValueError: could not broadcast input array from shape (5) into shape (92)

Я понимаю, что... Numpy занимает только первый ряд (5 элементов) и пытается транслировать в столбец из 92 элементов.

Итак... теперь мне интересно, как получить первый элемент каждого списка в столбце из 92 элементов, Scratchin моя голова.... Заранее спасибо за совет.

Answer 1

Я думаю, что recarray имеет dtype где одно из полей имеет форму 5:

In [48]: dt = np.dtype([('X',int,5),('Y',float)])
In [49]: arr = np.zeros(3, dtype=dt)
In [50]: arr
Out[50]: 
array([([0, 0, 0, 0, 0], 0.), ([0, 0, 0, 0, 0], 0.),
       ([0, 0, 0, 0, 0], 0.)], dtype=[('X', '<i8', (5,)), ('Y', '<f8')])

Доступ к этому полю по имени дает массив, который имеет (3,5) форму (аналог вашей (92,5):

In [51]: arr['X']
Out[51]: 
array([[0, 0, 0, 0, 0],
       [0, 0, 0, 0, 0],
       [0, 0, 0, 0, 0]])

Это можно описать как список из 5 элементов для каждой записи, но при индексировании по имени поля создается двумерный массив, который может индексироваться как любой двумерный массив.

Давайте установим эти значения в нечто интересное:

In [52]: arr['X'] = np.arange(15).reshape(3,5)
In [53]: arr
Out[53]: 
array([([ 0,  1,  2,  3,  4], 0.), ([ 5,  6,  7,  8,  9], 0.),
       ([10, 11, 12, 13, 14], 0.)],
      dtype=[('X', '<i8', (5,)), ('Y', '<f8')])

Мы можем получить первый столбец этого поля с помощью:

In [54]: arr['X'][:,0]
Out[54]: array([ 0,  5, 10])

Если у вас есть несколько полей с такой структурой, вам, вероятно, придется обращаться к каждому из них по имени. Существует ограничение на то, что вы можете сделать с многопольным индексированием.