要按照存储在列中的日期列表筛选数据框,可以使用以下代码示例:
import pandas as pd
# 创建示例数据框
data = {'日期': ['2020-01-01', '2020-01-02', '2020-01-03', '2020-01-04', '2020-01-05'],
'数值': [1, 2, 3, 4, 5]}
df = pd.DataFrame(data)
# 将日期列转换为日期类型
df['日期'] = pd.to_datetime(df['日期'])
# 创建日期列表
date_list = ['2020-01-02', '2020-01-04']
# 使用isin方法筛选数据框
filtered_df = df[df['日期'].isin(pd.to_datetime(date_list))]
print(filtered_df)
输出结果为:
日期 数值
1 2020-01-02 2
3 2020-01-04 4
在示例中,首先将日期列转换为日期类型,然后创建日期列表。接下来,使用isin
方法将筛选出与日期列表中的日期匹配的行,并将结果存储在filtered_df
中。最后,打印筛选后的数据框。