使用 multiIndex 在 pandas 数据框中获取未排序的列级别

Question

背景：

我有一个带有 MultiIndex 的 pandas 数据框。

我想得到未排序的 columns.levels。​​

df.column.levels 将它们提供给我，但已排序。

示例：

worms=['worm1', 'worm2', 'worm3']
bodyparts=['head', 'vulva', 'tail']
coords=['x', 'y']
arrays=pd.MultiIndex.from_product([worms, bodyparts, coords],

                           names=['worms', 'bodyparts', 'coords'])


idx = pd.Index(np.arange(0,5000),name='frames')
df=pd.DataFrame(index=idx, columns=arrays)

看起来像这样：

worms   worm1   worm2   worm3
bodyparts   head    vulva   tail    head    vulva   tail    head    vulva   tail
coords  x   y   x   y   x   y   x   y   x   y   x   y   x   y   x   y   x   y
frames                                                                      
0   NaN     NaN     NaN     NaN     NaN     NaN     NaN     NaN     NaN     NaN     NaN     NaN     NaN     NaN     NaN     NaN     NaN     NaN
1   NaN     NaN     NaN     NaN     NaN     NaN     NaN     NaN     NaN     NaN     NaN     NaN     NaN     NaN     NaN     NaN     NaN     NaN

现在如果我这样做：

df.columns.levels[1]

我明白了：

Index(['head', 'tail', 'vulva'], dtype='object', name='bodyparts')

我想要什么：

Index(['head', 'vulva', 'tail'], dtype='object', name='bodyparts')

有什么选择吗？

Answer 1

这会按出现顺序为您提供它们：

pd.unique(df.columns.to_frame()['bodyparts'])

输出：['head' 'vulva' 'tail']