如何將 dataframe 轉換為 numpy 數組？

Question

所以我目前正在制作一個神經網絡 MLP (Multi-layer-Perceptron) 以秒為單位對海灘到大海之間的海龜速度進行分組分類，看起來有點像這樣。

現在在我的 Jupyter Notebook 上輸入以下內容

data = pd.read_csv("SeaTurtles.csv")
data

它顯示了 dataframe 中的數據

我想要做的是將這兩組分開，即“培訓”數據的 A 組和“測試”數據的 B 組。 我想將 dataframe 轉換為 NumPy 數組，以便我更容易將它們分類到 SpeciesCode 中。 所以我輸入：

newdf=data.drop(['Group1','Group 2'], axis = 1)
newdf

我需要把它分成兩個測試組 A 和 B 所以我輸入

groupA=newdf.loc[newdf['Test*'] == 'A']
groupB=newdf.loc[newdf['Test*'] == 'B']

B組

我所做的是將這些轉換為 numpy 數組 with.to_numpy()

groupA = groupA.to_numpy()
groupA

它返回“AttributeError：'numpy.ndarray' object has no attribute 'to_numpy'”

我的問題是我做錯了什么還是有另一種方法可以將這個 dataframe 轉換為 numpy 數組，以便我可以開始訓練數據？ 先感謝您。

Answer 1

Pandas dataframe 是一種二維數據結構，用於以行和列格式存儲和檢索數據。

您可以使用 df.to_numpy() 方法將 pandas dataframe 轉換為 numpy 數組。

您可以使用以下代碼片段將 pandas dataframe 轉換為 numpy 數組。 numpy_array = df.to_numpy() print(type(numpy_array))

Output：

<類'numpy.ndarray'>