将二维数组转换为一维数组以创建数据框

Question

我有三个 2D np.array，它们在数学上是 [8:1550] 矩阵，我想将它们表示为 12400 个数字（8 x 1550 = 12400 ...）的 1D np.array，以便我可以稍后创建一个 DataFrame这段代码：

Exported_Data = pd.DataFrame({"UD": UD_Data, "NS": NS_Data, "EW": EW_Data})
Exported_Data.to_csv("EXCEL.csv")

举一个更简单的例子，如果我有这个：

A = np.array([[1, 2, 3],
              [4, 5, 6],
              [7, 8, 9]])

我想从中获得这个：

B = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9])

最好的方法是什么？

Answer 1

您可以使用A.flatten()将二维数组转换为一维数组。

Answer 2

我建议使用reshape 。 它最有可能创建一个视图并且效率更高，而np.flatten创建一个副本：

B = A.reshape(-1)

-1隐式处理所需的尺寸大小。