时间戳

首页 > 解决方案 > MultiIndex 的滚动级别

python - MultiIndex 的滚动级别

问题描述

假设我有一个数据框，其中我的列是 MultiIndex

col = pd.MultiIndex.from_product(
    [[1, 2], ['A', 'B'], ['First', 'Second']],
    names=['Cat', 'Dog', 'Bird']
)
dat = np.arange(16).reshape(2, -1)
df = pd.DataFrame(dat, columns=col)
df

Cat      1                         2                    
Dog      A            B            A            B       
Bird First Second First Second First Second First Second
0        0      1     2      3     4      5     6      7
1        8      9    10     11    12     13    14     15

我想调整列，使Bird级别在顶部，Cat级别向下移动到中间，然后Dog到底部。

尝试 1

Usingswaplevel可以连续使用，但是在中间创建一个完整的数据框只是为了增加列感觉很笨拙。

df.swaplevel(0, 2, 1).swaplevel(1, 2, 1).sort_index(1)

Bird First             Second            
Cat      1       2          1       2    
Dog      A   B   A   B      A   B   A   B
0        0   2   4   6      1   3   5   7
1        8  10  12  14      9  11  13  15

尝试 2

创建新的MultiIndex应该是高效的，但不是那么直观，并且可能不必要地冗长。

def roll(x):
    return x[-1:] + x[:-1]

df.set_axis(
    pd.MultiIndex.from_tuples(
        [roll(x) for x in df.columns.values],
        names=roll(df.columns.names)
    ), axis=1, inplace=False).sort_index(1)

Bird First             Second            
Cat      1       2          1       2    
Dog      A   B   A   B      A   B   A   B
0        0   2   4   6      1   3   5   7
1        8  10  12  14      9  11  13  15

问题

是否有一种简洁直观的方法可以做到这一点，而无需在中间创建中间数据框？

标签： pythonpandasmulti-index

解决方案

这是你需要的先生吗？使用reorder_levels：

df.reorder_levels(['Bird','Cat','Dog'],axis=1).sort_index(level=0,axis=1)
Out[396]: 
Bird First             Second            
Cat      1       2          1       2    
Dog      A   B   A   B      A   B   A   B
0        0   2   4   6      1   3   5   7
1        8  10  12  14      9  11  13  15

推荐阅读