Изменить размер массива в Python (Numpy)

Question

У меня есть numpy массив формы arr.shape: (416809, )

, который имеет:

arr[0].shape:
(300,)

и я, sh, чтобы изменить форму : (416809, 300)

Я искал в документации, но не смог найти решение.

Любая помощь будет признательна!

Спасибо!

Answer 1

Чтобы преобразовать форму NumPy array ndarray, используйте метод reshape() из ndarray или функцию numpy.reshape().

Вот пример для справки

import numpy as np

a = np.arange(24)

print(a)
# [ 0  1  2  3  4  5  6  7  8  9 10 11 12 13 14 15 16 17 18 19 20 21 22 23]

print(a.shape)
# (24,)

a_4_6 = a.reshape([4, 6])

print(a_4_6)
# [[ 0  1  2  3  4  5]
#  [ 6  7  8  9 10 11]
#  [12 13 14 15 16 17]
#  [18 19 20 21 22 23]]