有没有更好的方法来展平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)