使用 multiIndex 在 Pandas 數據框中獲取未排序的列級別

Question

背景：

我有一個帶有 MultiIndex 的 Pandas 數據框。

我想讓 column.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']