python - 将数组从 2d reshape 为 3d

Question

我目前在将 numpy.ndarray 从 2d reshape 为 3d 时遇到问题..

我的 numpy.ndarray 的当前形状是 (221286, 2050)，我需要它是 (221286, 1, 2050)

我尝试这样做:

train_set_data_vstacked_normalized_reshaped = np.reshape(train_set_data_vstacked_normalized.shape[0],1,train_set_data_vstacked_normalized.shape[1])

但这似乎创建了一种不同的 numpy.ndarray....

Answer 1

np.reshape 当用作函数时，将要 reshape 的数组作为第一个参数，将新形状作为第二个参数。所以这应该可以做到:

shape = your_long_named_array.shape
your_long_named_array_reshaped = np.reshape(your_long_named_array,
                                            (shape[0], 1, shape[1]))

您还可以使用 ndarrays 的 .reshape 方法，该方法不需要您显式地将形状包装在元组中:

your_long_named_array_reshaped = your_long_named_array.reshape(shape[0], 1,
                                                               shape[1])

尽管对于这种特殊情况，最方便的可能是使用 np.newaxis 索引数组:

your_long_named_array_reshaped = your_long_named_array[:, np.newaxis, :]