как сделать массив из массивов numpy

Аватар пользователя Maksim Litvinov
Maksim Litvinov
10 июля 2024

Для создания массива из массивов в библиотеке NumPy вы можете использовать функцию numpy.array() с указанием списка списков в качестве аргумента. Например, чтобы создать массив из двух массивов:

import numpy as np

arr1 = np.array([1, 2, 3])
arr2 = np.array([4, 5, 6])

arr_of_arrs = np.array([arr1, arr2])

print(arr_of_arrs)

Этот код создаст массив arr_of_arrs, содержащий массивы arr1 и arr2.

Если вы хотите создать массив из массивов с различными формами (разной длины), вам нужно будет использовать функцию numpy.vstack():

import numpy as np

arr1 = np.array([1, 2, 3])
arr2 = np.array([4, 5, 6, 7])

arr_of_arrs = np.vstack((arr1, arr2))

print(arr_of_arrs)

Обратите внимание, что массивы должны иметь одинаковую длину вдоль оси, по которой вы объединяете их. Если массивы имеют разные размеры вдоль этой оси, вы получите ошибку.

1 0