python - Pandas 按日期过滤值

Question

我有一个日期数组 dateref 并想要获取 pandas 数据帧数据的值按 dateref 过滤(仅包含 dateref 中的日期值):

我尝试过这个，但它不起作用:

  df: Out[36]: rec.array([ (datetime.date(2007, 4, 10), 105), 

  df[df.date== dateref ]

data.date:日期数组

data.value:值数组

编辑:此返回错误

data[data.date.isin(dateref)] 
AttributeError: 'numpy.ndarray' object has no attribute 'isin'

Answer 1

我认为你需要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)]