Как сгладить только некоторые размеры массива numpy

Question

Есть ли быстрый способ «сгладить» или сгладить только некоторые из первых измерений в массиве numpy?

Например, учитывая NumPy массив размеров (50,100,25), размеры результирующие будет (5000,25)

Answer 1

Посмотрите на numpy.reshape.

>>> arr = numpy.zeros((50,100,25)) 
>>> arr.shape 
# (50, 100, 25) 

>>> new_arr = arr.reshape(5000,25) 
>>> new_arr.shape 
# (5000, 25) 

# One shape dimension can be -1. 
# In this case, the value is inferred from 
# the length of the array and remaining dimensions. 
>>> another_arr = arr.reshape(-1, arr.shape[-1]) 
>>> another_arr.shape 
# (5000, 25) 

Answer 2

Небольшое обобщение ответа Александра - np.reshape может принимать -1 в качестве аргумента, что означает «общий размер массива делится на продукт всех других перечисленных размеров»:

например чтобы сгладить все, кроме последнего измерения:

>>> arr = numpy.zeros((50,100,25)) 
>>> new_arr = arr.reshape(-1, arr.shape[-1]) 
>>> new_arr.shape 
# (5000, 25) 

Answer 3

Небольшое обобщение на ответ Петра - вы можете указать диапазон над формой исходного массива, если вы хотите выйти за пределы трех одномерных массивов.

например. выравниваться все, но последние два размеров:

arr = numpy.zeros((3, 4, 5, 6)) 
new_arr = arr.reshape(-1, *arr.shape[-2:]) 
new_arr.shape 
# (12, 5, 6) 

EDIT: небольшое обобщение моего предыдущего ответа - вы можете, конечно, также указать диапазон в начале из RESHAPE тоже:

arr = numpy.zeros((3, 4, 5, 6, 7, 8)) 
new_arr = arr.reshape(*arr.shape[:2], -1, *arr.shape[-2:]) 
new_arr.shape 
# (3, 4, 30, 7, 8)