python - 从带有日期的 Pandas 数据框中选择行

Question

给定一个简单的数据框

df = pd.DataFrame(np.random.rand(5,3))

我可以使用标签 1 和 3 来选择记录

df.loc[[1,3]]

但是，如果我更改索引以便它使用日期...

df.index = pd.date_range('1/1/2010', periods=5)

这不再有效:

df.loc[['2010-01-02', '2010-01-04']]

KeyError: "None of [['2010-01-02', '2010-01-04']] are in the [index]"

.loc 在此上下文中如何与日期一起使用？

Answer 1

一种可能的解决方案是将日期转换为 DatetimeIndex或 to_datetime然后效果很好:

print (df.loc[pd.DatetimeIndex(['2010-01-02', '2010-01-04'])])

                   0         1         2
2010-01-02  0.827821  0.285281  0.781960
2010-01-04  0.872664  0.895636  0.368673

---

print (df.loc[pd.to_datetime(['2010-01-02', '2010-01-04'])])

                   0         1         2
2010-01-02  0.218419  0.806795  0.454356
2010-01-04  0.038826  0.741220  0.732816