python - Numpy 将N维数组转换为二维数组

Question

给定一个形状为(N,d_1,...,d_{n-1})的n维数组，我想将该数组转换为形状为二维数组(N,D)，其中 D =\prod_i d_i。 numpy 中有什么巧妙的技巧可以做到这一点吗？

Answer 1

您可以使用numpy.reshape()方法 ，以及 array.shape获取原始数组的形状和 np.prod()取 d_1 , d_2 , ... 的乘积。示例-

In [22]: import numpy as np

In [23]: na = np.arange(10000).reshape((5,2,10,10,10))

In [26]: new_na = np.reshape(na, (na.shape[0], np.prod(na.shape[1:])))

In [27]: new_na.shape
Out[27]: (5, 2000)

或者@hpaulj 在评论中建议的更简单的版本 -

In [31]: new_na = np.reshape(na, (na.shape[0], -1))

In [32]: new_na.shape
Out[32]: (5, 2000)