Несколько полей сразу в NumPy

Если передать список имён полей, вернётся новый структурированный массив только с выбранными столбцами.

Выберем имя и возраст вместе:

import numpy as np

dtype = np.dtype([('name', 'U10'), ('age', 'i4')])
people = np.array([('Ann', 20), ('Bob', 31)], dtype=dtype)
subset = people[['name', 'age']]
print(subset)

Порядок полей в результате совпадает с порядком имён в списке:

import numpy as np

dtype = np.dtype([('name', 'U10'), ('age', 'i4')])
people = np.array([('Ann', 20), ('Bob', 31)], dtype=dtype)
subset = people[['age', 'name']]
print(subset.dtype.names) # выведет ('age', 'name')

Создайте массив Ann, 20 и Bob, 31 с полями name и age. Получите подмассив с обоими полями и выведите его.

Создайте массив из Kim, 25 и Lee, 40 с теми же полями. Выберите поля age и name и выведите имена полей в результате.