python - 根据特定日期条件在Python中选择日期列

Question

这是我的示例代码。我的数据库包含一年中每个日期的列，可以追溯到多年前。每列对应一个特定日期。

import pandas as pd
df = pd.DataFrame([[10, 5, 25, 67,25,56], 
                   [20, 10, 26, 45, 56, 34], 
                   [30, 3, 27, 34, 78, 34], 
                   [40, 9, 28, 45, 34,76]], 
                  columns=[pd.to_datetime('2022-09-14'), pd.to_datetime('2022-08-14'), pd.to_datetime('2022-07-14'), pd.to_datetime('2021-09-14'),
                              pd.to_datetime('2020-09-14'), pd.to_datetime('2019-09-14')])

有没有办法根据年、月或季度仅选择符合特定条件的列。

例如，我希望每年只获取与今天(任何开始日期)相同的日期的列。例如，今天是 2022 年 9 月 14 日，我只需要 2021 年 9 月 14 日、2020 年 9 月 14 日等的列。另一种选择可能是按月或按季度执行相同的操作。 在 pandas 中如何做到这一点？

Answer 1

是的，你可以这样做:

# day
df.loc[:, df.columns.day == 14]

   2022-09-14  2022-08-14  2022-07-14  2021-09-14  2020-09-14  2019-09-14
0          10           5          25          67          25          56
1          20          10          26          45          56          34
2          30           3          27          34          78          34
3          40           9          28          45          34          76


# month
df.loc[:, df.columns.month == 9]

   2022-09-14  2021-09-14  2020-09-14  2019-09-14
0          10          67          25          56
1          20          45          56          34
2          30          34          78          34
3          40          45          34          76


# quarter
df.loc[:, df.columns.quarter == 3]

   2022-09-14  2022-08-14  2022-07-14  2021-09-14  2020-09-14  2019-09-14
0          10           5          25          67          25          56
1          20          10          26          45          56          34
2          30           3          27          34          78          34
3          40           9          28          45          34          76