我是 Python 新手。我想知道如何在 Python 中提取列的前三个值。我使用了 CSV 文件。这是我的代码:
import pandas as pd
census_df = pd.read_csv('census.csv')
def answer_five():
return census_df['COUNTY'].idxmax(3)
answer_five()
Run Code Online (Sandbox Code Playgroud)
你想要
# get first three
census_df['COUNTY'].head(3)
Run Code Online (Sandbox Code Playgroud)
或者
# get largest three
census_df['COUNTY'].nlargest(3)
Run Code Online (Sandbox Code Playgroud)
如果你想用3个最大的记录COUNTY值
census_df.iloc[census_df['COUNTY'].argsort()[-3:]]
Run Code Online (Sandbox Code Playgroud)