python - 查找 Pandas 日期时间索引中是否存在指定月份字符串中的至少一个条目

Question

我有一个 pandas 数据框 test_df ，如下所示:

                            reading                               
01-Jan-2016 00:00:00.000  20.464020
02-Jan-2016 00:00:00.000  22.440950
03-Jan-2016 00:00:00.000  27.181500
04-Jan-2016 00:00:00.000  25.318260
05-Jan-2016 00:00:00.000  25.376050
06-Jan-2016 00:00:00.000   0.067112
07-Jan-2016 00:00:00.000  19.313950
08-Jan-2016 00:00:00.000  26.677340
09-Jan-2016 00:00:00.000  26.801620
10-Jan-2016 00:00:00.000  22.583950
11-Jan-2016 00:00:00.000   0.002765
12-Jan-2016 00:00:00.000  26.496440
13-Jan-2016 00:00:00.000  23.233720
14-Jan-2016 00:00:00.000  23.956080
15-Jan-2016 00:00:00.000  26.958120
16-Jan-2016 00:00:00.000  27.351270
17-Jan-2016 00:00:00.000  28.348710
18-Jan-2016 00:00:00.000  25.494090
19-Jan-2016 00:00:00.000  26.342880
20-Jan-2016 00:00:00.000  24.645530

问题:给定一个像“2016-01”又名“yyyy-mm”这样的字符串，我想找出指定月份的任何条目是否存在于 pandas 数据帧的索引中test_df。

我期望“2016-01”为 True，对于任何其他字符串为 False。寻找最简洁的方法来做到这一点。

问题设置:

为了简单起见，这是获取测试数据帧的代码:

import pandas as pd
temp_df = pd.read_json('{"reading":{"01-Jan-2016 00:00:00.000":20.46402,"02-Jan-2016 00:00:00.000":22.44095,"03-Jan-2016 00:00:00.000":27.1815,"04-Jan-2016 00:00:00.000":25.31826,"05-Jan-2016 00:00:00.000":25.37605,"06-Jan-2016 00:00:00.000":0.06711243,"07-Jan-2016 00:00:00.000":19.31395,"08-Jan-2016 00:00:00.000":26.67734,"09-Jan-2016 00:00:00.000":26.80162,"10-Jan-2016 00:00:00.000":22.58395,"11-Jan-2016 00:00:00.000":0.002765084,"12-Jan-2016 00:00:00.000":26.49644,"13-Jan-2016 00:00:00.000":23.23372,"14-Jan-2016 00:00:00.000":23.95608,"15-Jan-2016 00:00:00.000":26.95812,"16-Jan-2016 00:00:00.000":27.35127,"17-Jan-2016 00:00:00.000":28.34871,"18-Jan-2016 00:00:00.000":25.49409,"19-Jan-2016 00:00:00.000":26.34288,"20-Jan-2016 00:00:00.000":24.64553}}')

我已经尝试过:

>>'2016-01' in test_df.index
False

Answer 1

如果使用DatetimeIndex，您可以使用 to_period用于转换为 PeriodIndex 然后 any (谢谢John Zwinck ):

print (temp_df.index.to_period('m'))
PeriodIndex(['2016-01', '2016-01', '2016-01', '2016-01', '2016-01', '2016-01',
             '2016-01', '2016-01', '2016-01', '2016-01', '2016-01', '2016-01',
             '2016-01', '2016-01', '2016-01', '2016-01', '2016-01', '2016-01',
             '2016-01', '2016-01'],
            dtype='period[M]', freq='M')

print (temp_df.index.to_period('m') == '2016-01')
[ True  True  True  True  True  True  True  True  True  True  True  True
  True  True  True  True  True  True  True  True]

print ((temp_df.index.to_period('m') == '2016-01').any())
True