要按照特定的年份筛选DataFrame中的某一列,可以使用如下代码:
import pandas as pd
# 生成测试数据
data = {'year': [2010, 2010, 2011, 2011, 2012, 2012],
'value': [1, 2, 3, 4, 5, 6]}
df = pd.DataFrame(data)
# 按照年份筛选value列
selected_year = 2011
filtered_df = df[df['year'] == selected_year]['value']
# 打印结果
print(filtered_df)
此代码将输出:
2 3
3 4
Name: value, dtype: int64
这个结果是一个Series对象,它包含所有年份为2011的value列的值。如果想要输出整个DataFrame而不仅仅是一个Series对象,可以直接像下面这样做:
filtered_df = df[df['year'] == selected_year]
print(filtered_df)
这将输出:
year value
2 2011 3
3 2011 4