將 dataframe 列添加到 numpy.array

Question

對於回歸，我想將 dataframe 列添加到包含虛擬變量的 numpy.array。

目前，該數組如下所示：

[[0 0 0 0]
 [0 0 0 0]
 [0 0 0 0]
 [0 0 0 0]
 [0 0 0 0]
 [0 0 0 1]
 [0 0 1 0]]

我想添加 dataframe 列值（在此示例中有 7 行），以便方括號內有 5 個值（來自 dataframe 列的值和四個虛擬變量）。

有誰知道如何解決這個問題？

Answer 1

您可以使用 numpy 數組創建一個 dataframe：

array=np.array([[0, 0, 0, 0],[0, 0, 0, 0],[0, 0, 0, 0],[0, 0, 0, 0],[0, 0, 0, 0],[0, 0, 0, 1],[0, 0, 1, 0]])
new_dataframe = pd.DataFrame(data=array)

然后像這樣將您的列添加到其中：

new_dataframe['4'] = your_dataframe['column_name']