Pandas过滤具有特定年份的数据帧行

Question

我有一个数据框df,它有一个Date列.我想创建两个新的数据框架.一个包含df年份等于的所有行some_year,另一个包含df年份不等于的所有行的数据框some_year.我知道你可以做df.ix['2000-1-1' : '2001-1-1']但是为了获得2000年以来的所有行需要创建2个额外的数据帧然后连接/加入它们.

有这样的方式吗？

include = df[df.Date.year == year]
exclude = df[df['Date'].year != year]

这段代码不起作用,但有没有类似的方式？

Answer 1

您可以使用datetime访问者.

import datetime as dt
df['Date'] = pd.to_datetime(df['Date'])

include = df[df['Date'].dt.year == year]
exclude = df[df['Date'].dt.year != year]

Answer 2

您可以通过反向掩码简化它~和条件使用Series.dt.year与int流延字符串year：

mask = df['Date'].dt.year == int(year)
include = df[mask]
exclude = df[~mask]