Как получить массив с смешанными типами?

Question

Я хотел бы создать массив со смешанными типами.Другие SO вопросы, которые я обнаружил, создают либо массив на основе object, либо вложенный массив.

И то, и другое я не хочу.

Как будет выглядеть синтаксис для создания массива с пустым массивомодин str и два int столбца?

Это мой настоящий код:

import numpy as np

b = np.empty((0, 3), )
b = np.insert(b, b.shape[0], [[1, 2, 3]], axis=0)
b = np.insert(b, b.shape[0], [[1, 2, 3]], axis=0)

print(b)
print("---")

a = np.empty((0, 3), dtype='S4, int, int')
a = np.insert(a, a.shape[0], ("a", 2, 3), axis=0)
a = np.insert(a, a.shape[0], ("a", 2, 3), axis=0)

print(a)

Вывод:

[[1. 2. 3.]
 [1. 2. 3.]]
---
[[(b'a', 2, 3) (b'a', 2, 3) (b'a', 2, 3)]
 [(b'a', 2, 3) (b'a', 2, 3) (b'a', 2, 3)]]

РЕДАКТИРОВАТЬ:

И что мне нужно для массива a:

[["a" 2 3]
 ["a" 2 3]]

Answer 1

Ваш второй массив близок, хотя я бы сделал это с помощью индексации, а не вставки (что медленнее):

In [431]: a = np.zeros(3, dtype='S4, int, int')
In [432]: a[0] = ('a', 2, 3)
In [433]: a[1] = 1
In [434]: a
Out[434]: 
array([(b'a', 2, 3), (b'1', 1, 1), (b'', 0, 0)],
      dtype=[('f0', 'S4'), ('f1', '<i8'), ('f2', '<i8')])

Список кортежей также является хорошим способом создания такого массива:

In [436]: a = np.array([('a',2,3),('b',4,5)], dtype='S4, int, int')
In [437]: a
Out[437]: 
array([(b'a', 2, 3), (b'b', 4, 5)],
      dtype=[('f0', 'S4'), ('f1', '<i8'), ('f2', '<i8')])

Обратите внимание, что форма 1d (n,) с 3 полями.Поля не считаются измерениями.

Доступ к полям осуществляется по имени, а не по номеру столбца:

In [438]: a['f1']
Out[438]: array([2, 4])

Вы создали (2,3) массив и заполнили каждый«грести» с тем же.Вот почему у вас есть повторы, а у меня нет.

Со строкой Юникода dtype (по умолчанию для Py3):

In [439]: a = np.array([('a',2,3),('b',4,5)], dtype='U4, int, int')
In [440]: a
Out[440]: 
array([('a', 2, 3), ('b', 4, 5)],
      dtype=[('f0', '<U4'), ('f1', '<i8'), ('f2', '<i8')])
In [441]: print(a)
[('a', 2, 3) ('b', 4, 5)]