按列级别/轴1多索引选择Pandas数据

Question

这是一段代码，它完成了我尝试做的部分工作：

selection=df.xs(('year2012','3quarter','A'),level=[0,1,2],axis=1)

如何在轴1的每个级别中选择多个条件？ 以下似乎是完成此任务的错误方法。

selection=df.xs(( ['year2012','year2015'] , ['3quarter','4quarter'] , 'A'), level=[0,1,2], axis=1）

谢谢

Answer 1

考虑示例数据框df

mux = pd.MultiIndex.from_product([
        ['year{}'.format(i) for i in range(2010, 2017)],
        ['{}quarter'.format(i) for i in range(1, 5)],
        list('ABCD')
    ])

df = pd.DataFrame([np.arange(len(mux))], columns=mux)

df

执行此操作的方法是使用pd.IndexSlice

l1 = ['year2012','year2015']
l2 = ['3quarter','4quarter']
l3 = 'A'

df.loc[:, pd.IndexSlice[l1, l2, l3]]