Есть ли лучший способ сгладить часть массива NumPy?

Question

Я пытаюсь сгладить часть массива NumPy.

вот исходный набор данных

dataset = np.random.random((999,5,5))

этот код выравнивает топ 99 примеров

flatten_list = []
for i in range(99):
    flatten_list.append(dataset[i].flatten())
flatten_arr = np.array(flatten_list)

хотя этот код работает, есть ли лучший способ сделать это?

уведомление, flatten_arr имеет форму (99, 25), а не (2475,)

Answer 1

это может удовлетворить вас нужно

flatten_arr = dataset[:99].reshape(99, 25)