将熊猫中的一行输出到一个数组

Question

我希望能够拔出一个任意行，例如第四行，并获得一个数组，以便我可以通过另一个功能发送它。

最简单的方法是什么？

Answer 1

由于熊猫数据在内部存储为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

请记住，如果您的数据框中包含多种类型，则行态Numpy数组的dtype可能会变成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