且构网

分享程序员开发的那些事...
且构网 - 分享程序员编程开发的那些事

pandas 按日期过滤值

更新时间:2023-11-26 14:02:22

我认为您需要

I think you need numpy.in1d:

data[np.in1d(data.date, dateref)] 

示例:

import numpy as np
import datetime as datetime

test = np.array([datetime.date(2007, 4, 10), 
                 datetime.date(2007, 5, 10), 
                 datetime.date(2007, 6, 10)])

states = [datetime.date(2007, 4, 10), 
          datetime.date(2007, 5, 10)]

mask = np.in1d(test, states)
print mask
[ True  True False]

print test[mask]
[datetime.date(2007, 4, 10) datetime.date(2007, 5, 10)]