将 (8750, 4, 35) np 数组压缩为 (35000, 35)

Question

我有一个形状为 (8750, 4, 35) 的 numpy 数组，我试图将其压缩到 (35000, 35)。我试过 np.squeeze:

a = np.array(pred_f)
np.squeeze(a).shape

但这仍然会产生 (8750, 4, 35) 并且更改轴参数会导致错误：

ValueError: cannot select an axis to squeeze out which has size not equal to one

有人知道我该怎么做吗？

Answer 1

import numpy as np
a = np.random.random((8750, 4, 35))
a = np.reshape(a,(-1,35))
print(a.shape) #(35000,35)

如 Dr.Snoopy 所述，reshape 就是您要查找的内容。它看起来像这样。