我想要能够提取任意一行,比如第4行,并且获得一个数组,以便我可以将其发送到另一个函数中。
最简单的方法是什么?
由于Pandas数据内部存储为Numpy数组,您可以直接提取Numpy表示。
使用pd.Series.to_numpy
方法:
df.iloc[3].to_numpy() # output 4th row as Numpy array
请使用 pd.Series.values
属性:
df.iloc[3].values # output 4th row as Numpy array
请记住,如果您的数据帧包含多种类型,则行向Numpy数组的dtype
可能会变为object
。这将导致后续操作效率低下。
# Dummy DataFrame
df = pd.DataFrame({'col1': [1,2,3], 'col2': [4,5,6]})
# Extract second row (index: 1)
df.iloc[1].values