有没有一种方法可以从pandas DataFrame的特定列中生成一个numpy数组？

Question

这是在构建数据

X = np.array([[1.   , 0.697, 0.46 ],
       [2.   , 0.774, 0.376],
       [3.   , 0.634, 0.264]])
df = pd.DataFrame(X, columns=list('ABC'))

numpy切片

X[:,1:]

输出（snippet_1）

array([[0.697, 0.46 ],
       [0.774, 0.376],
       [0.634, 0.264]])

假设我们以df而不是X开头

df.to_numpy()

输出

array([[1.   , 0.697, 0.46 ],
       [2.   , 0.774, 0.376],
       [3.   , 0.634, 0.264]])

有没有一种方法可以从snippet_1之类的pandas DataFrame的后两列生成一个numpy数组？

Answer 1

使用iloc选择列

df.iloc[:,1:].to_numpy()

array([[0.697, 0.46 ],
       [0.774, 0.376],
       [0.634, 0.264]])