假设我有一个2D数组，并且我想将其重塑为1D，每行一个值

问题描述

这是我的数组

arr = np.array([[0, 1],
[3, 4],
[6, 7]])

flat_arr = np.reshape(arr, -1)

我得到以下结果：

[0 1 2 3 4 5 6 7 8]

我希望得到的结果是：

[0]
[1]
[3]
[4]
[5]...

---

解决方案

有几种方法：

flat_arr[:, None]
flat_arr[:, np.newaxis]
np.expand_dims(flat_arr, axis=1)

此外，您还可以按如下方式重塑它：

arr.reshape(-1, 1)