将 Pandas 选定的列转换为 2D numpy 数组

Question

我有一个包含 10 列和 N 行的 Pandas 数据框。我想从我的数据框中选择两列并将它们放入维度为 (N, 2) 的 NumPy 二维数组中。怎么做？

Answer 1

有一种方法可以将数据帧转换为 numpy 数组：https : //pandas.pydata.org/pandas-docs/stable/reference/api/pandas.DataFrame.to_numpy.html

你可以只选择你想要的列：

df[['Col1','Col2']].to_numpy()