有沒有一種方法可以從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]])