Criação de array com um único elemento estruturado contendo um array

Oct 06 2020

Eu tenho um tipo de d como este:

>>> dt = np.dtype([('x', object, 3)])
>>> dt
dtype([('x', 'O', (3,))])

Um campo denominado 'x', contendo três ponteiros. Eu gostaria de construir uma matriz com um único elemento deste tipo:

>>> a = np.array([(['a', 'b', 'c'])], dtype=dt)
>>> b = np.array([(np.array(['a', 'b', 'c'], dtype=object))], dtype=dt)
>>> c = np.array((['a', 'b', 'c']), dtype=dt)
>>> d = np.array(['a', 'b', 'c'], dtype=dt)

>>> e = np.array([([['a', 'b', 'c']])], dtype=dt)

Todas as cinco declarações geram o mesmo resultado incorreto:

array([[(['a', 'a', 'a'],), (['b', 'b', 'b'],), (['c', 'c', 'c'],)]],
      dtype=[('x', 'O', (3,))])

Se eu tentar eliminar a lista / matriz interna, obtenho um erro:

>>> f = np.array([('a', 'b', 'c')], dtype=dt)
ValueError: could not assign tuple of length 3 to structure with 1 fields.

O mesmo erro acontece para

>>> g = np.array(('a', 'b', 'c'), dtype=dt)

Fiquei sem combinações possíveis para tentar. O resultado que procuro é

 array([(['a', 'b', 'c'],)], dtype=[('x', 'O', (3,))])

Como faço para criar uma matriz que tem um elemento do dtype especificado?

Até agora, a única abordagem que encontrei é a atribuição manual:

z = np.empty(1, dtype=dt)
z['x'][0, :] = ['a', 'b', 'c']

OU

z[0]['x'] = ['a', 'b', 'c']

Isso parece uma solução alternativa desnecessária para algo que np.arraydeveria ser capaz de lidar imediatamente.

Respostas

1 hpaulj Oct 05 2020 at 22:50
In [44]: dt = np.dtype([('x', object, 3)])   # corrected
In [45]: dt
Out[45]: dtype([('x', 'O', (3,))])
In [46]: np.empty(3, dt)
Out[46]: 
array([([None, None, None],), ([None, None, None],),
       ([None, None, None],)], dtype=[('x', 'O', (3,))])
In [47]: np.array([(['a','b','c'],)], dt)
Out[47]: array([(['a', 'b', 'c'],)], dtype=[('x', 'O', (3,))])

A formatação de entrada deve corresponder à formatação de saída.

In [48]: arr = np.empty(3, dt)
In [49]: arr['x']
Out[49]: 
array([[None, None, None],
       [None, None, None],
       [None, None, None]], dtype=object)
In [50]: arr['x'][0]
Out[50]: array([None, None, None], dtype=object)
In [51]: arr['x'][0] = ['a','b','c']
In [52]: arr
Out[52]: 
array([(['a', 'b', 'c'],), ([None, None, None],), ([None, None, None],)],
      dtype=[('x', 'O', (3,))])