將 Pandas 中的一行輸出到一個數組

Question

我希望能夠拉出任意行，例如第 4 行，並獲得一個數組，以便我可以通過另一個函數發送它。

什么是最簡單的方法來做到這一點？

Answer 1

由於 Pandas 數據在內部存儲為 Numpy 數組，因此您可以直接提取 Numpy 表示。

v0.24+

使用pd.Series.to_numpy方法：

df.iloc[3].to_numpy()  # output 4th row as Numpy array

v0.24之前

使用pd.Series.values屬性：

df.iloc[3].values  # output 4th row as Numpy array

只要記住，如果你的數據幀中包含多種類型的dtype在行numpy的陣列有可能成為object 。 這將導致后續操作效率低下。

Answer 2

# Dummy DataFrame
df = pd.DataFrame({'col1': [1,2,3], 'col2': [4,5,6]})
# Extract second row (index: 1)
df.iloc[1].values