отфильтровывать конкретные даты в списке из мультииндекса - PullRequest
Новая
       11

отфильтровывать конкретные даты в списке из мультииндекса

0 голосов
/ 02 декабря 2018

У меня есть многоиндексные данные, в которых я хотел бы отфильтровать список конкретных дат, например: 

  date_list=[Timestamp('2018-05-19 00:00:00'),
 Timestamp('2018-06-24 00:00:00'),
 Timestamp('2014-11-12 00:00:00'),
 Timestamp('2015-11-11 00:00:00'),
 Timestamp('2012-05-28 00:00:00'),
 Timestamp('2012-06-23 00:00:00')]

Я пытаюсь отфильтровать эти даты по следующим, но это не работает: 

df.iloc[df.index.get_level_values('Date') != date_list]

Может кто-нибудь, пожалуйста, помогите.

1 Ответ

0 голосов
/ 02 декабря 2018

Используйте Index.isin с инвертированной логической маской с помощью ~, iloc следует удалить, поскольку фильтр по boolean indexing:

Примечание:

Проверьте, если DatetimeIndex до фильтрации: 

print (df.index.get_level_values('Date'))

df1 = df[~df.index.get_level_values('Date').isin(date_list)]

Другое решение с drop с параметрами level и errors: 

df1 = df.drop(date_list, level='Date', errors='ignore')

Образец : 

df = pd.DataFrame({'Date':['2018-05-19','2014-11-10','2018-06-24','2014-11-13'],
                   'ID':[1,1,2,2],
                   'Val':list('abcd')})

df['Date'] = pd.to_datetime(df['Date'])
df = df.set_index(['Date','ID'])
print (df)
              Val
Date       ID    
2018-05-19 1    a
2014-11-10 1    b
2018-06-24 2    c
2014-11-13 2    d

date_list=[pd.Timestamp('2018-05-19 00:00:00'),
 pd.Timestamp('2018-06-24 00:00:00'),
 pd.Timestamp('2014-11-12 00:00:00'),
 pd.Timestamp('2015-11-11 00:00:00'),
 pd.Timestamp('2012-05-28 00:00:00'),
 pd.Timestamp('2012-06-23 00:00:00')] 

print (df.index.get_level_values('Date'))
DatetimeIndex(['2018-05-19', '2014-11-10', '2018-06-24', '2014-11-13'], 
              dtype='datetime64[ns]', name='Date', freq=None)

df1 = df[~df.index.get_level_values('Date').isin(date_list)]
print (df1)
              Val
Date       ID    
2014-11-10 1    b
2014-11-13 2    d

df1 = df.drop(date_list, level='Date', errors='ignore')
print (df1)
              Val
Date       ID    
2014-11-10 1    b
2014-11-13 2    d
...