如何检查是否在大熊猫多指标

Question

存在一列比方说，我有这样一个多指标一个数据帧列：

In [29]: df = pd.DataFrame([[0] * 8], columns = pd.MultiIndex.from_product(
    [['a', 'b'], [1, 2], [2000, 2001]]) 
) 

In [30]: df 
Out[30]: 
    a     b 
    1   2   1   2 
    2000 2001 2000 2001 2000 2001 2000 2001 
0 0 0 0 0 0 0 0 0 

In [46]: df.columns.levels 
Out[46]: FrozenList([[u'a', u'b'], [1, 2], [2000, 2001]]) 

我需要知道，为0级的所有值和一些特定的值级别1的所有现有唯一值是什么（例如，DataFrame经历某个过程，其中对于一级和零级的一些值，级别2被丢弃）。迄今为止我所能得到的最好的是：

In [54]: level_1_val = 2 

In [55]: cols_series = df.columns.to_series() 

In [56]: cols_series[ 
    ....:  cols_series.index.get_level_values(1) == level_1_val 
    ....: ].index.get_level_values(2).unique() 

array([2000, 2001]) 

什么是更好的方法来做到这一点？

Answer 1

IIUC

df.xs(2, axis=1, level=1).groupby(axis=1, level=1).first().columns.values 

array([2000, 2001]) 

或者

df.xs(2, axis=1, level=1).columns.get_level_values(level=1).unique()