將一個 Pandas DataFrame 轉換為一個 NumPy 陣列
要將一個 Pandas DataFrame 轉換為一個 NumPy 陣列,我們可以使用 to_numpy()。
步驟
建立一個二維、大小可變、可能包含不同型別資料的表格資料,df。
列印輸入的 DataFrame。
使用 df.to_numpy() 列印給定陣列的 NumPy 陣列。
使用 df['x'].to_numpy() 列印給定陣列特定列的 NumPy 陣列。
示例
import pandas as pd df = pd.DataFrame( { "x": [5, 2, 1, 9], "y": [4, 1, 5, 10], "z": [4, 1, 5, 0] } ) print "Input DataFrame is:
", df print "DataFrame to numpy is:
", df.to_numpy() print "DataFrame to numpy is:
", df['x'].to_numpy()
輸出
Input DataFrame is: x y z 0 5 4 4 1 2 1 1 2 1 5 5 3 9 10 0 DataFrame to numpy is: [[ 5 4 4] [ 2 1 1] [ 1 5 5] [ 9 10 0]] DataFrame to numpy is: [5 2 1 9]
廣告