如何從 dataframe 創建 numpy 陣列？

Question

我正在使用包含 4 列的Pandas DataFrame：“TV”、“Radio”、“Newspaper”和“Sales”。 我需要創建兩個NumPy arrays： X包含前 3 列， y包含“銷售”列。 我發現有兩種方法可以從Pandas DataFrame 創建NumPy數組：

import numpy as np
X = np.array(adver_data.iloc[:,0:3].values)
y = np.array(adver_data["Sales"].values)

和：

import numpy as np
X = adver_data[['TV', 'Radio', 'Newspaper']].values
y = adver_data[['Sales']].values

問題是，即使這兩種方法實際上創建了一個NumPy數組，我的代碼也為這兩種方法給出了不同的結果，我不知道為什么，因為我看不出這兩種方法的 arrays 有何不同。

有什么區別？

Answer 1

假設您有“ n ”行數據。

編碼：

y = np.array(adver_data["Sales"].values)

給出y的形狀為(n,)即一維數組

而代碼：

y = adver_data[['Sales']].values

給出y的形狀為(n, 1)即二維數組

所以，你會得到不同的結果。