如何将 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'>