В рамках моего текущего проекта я использую сторонний пакет, который в какой-то момент ожидает массив формы (n,), который содержит n массивов формы (m_n, 2). Этот массив создается из списка массивов с помощью np.array (). В большинстве случаев можно ожидать, что подмассивы не будут иметь одинаковую форму во всех них. Это приводит к желаемому результату. Однако иногда все подмассивы имеют одинаковую форму. Затем np.array () возвращает вывод формы (n, m_n, 2). Есть ли способ заставить numpy дать мне вывод формы (n,)? Кроме того, я был бы очень благодарен за способ непосредственного создания массива, подобного этому.
Ниже приведен пример моей проблемы.
import numpy as np
a = np.zeros((3,2))
b = np.zeros((4,2))
list1 = [a, a]
list2 = [a, b]
array1 = np.array(list1)
array2 = np.array(list2)
print(np.shape(array1))
print(np.shape(array2))