python - 从 pandas DataFrame MultiIndex 中选择命名索引级别

Question

我创建了一个数据框:

df1 = pandas.read_csv(ifile_name,  header=None,  sep=r"\s+",  usecols=[0,1,2,3,4],
                              index_col=[0,1,2], names=["year", "month", "day", "something1", "something2"])

现在我想创建另一个数据框，其中 year>2008。因此我尝试了:

df2 = df1[df1.year>2008]

但是出现错误:

AttributeError: 'DataFrame' object has no attribute 'year'

我猜，它在列中看不到“年份”，因为我在 index.html 中定义了它。但是在这种情况下，我如何才能获取基于 year>2008 的数据呢？

Answer 1

使用 MultiIndex.get_level_values 按名称获取级别并为行选择创建 bool 掩码:

df2 = df1[df1.index.get_level_values('year') > 2008]

如果您打算进行修改，请创建 df1 的副本，以便不对 View 进行操作。

df2 = df1[df1.index.get_level_values('year') > 2008].copy()