[英]Reorder levels of MultiIndex in a pandas DataFrame
我有一個看起來像這樣的 DataFrame:
>>> df = pd.DataFrame(index=pd.MultiIndex.from_tuples([(num,letter,color)
for num in range(1,3)
for letter in ['a','b','c'] for color in ['Red','Green']],
names=['Number','Letter','Color']))
>>> df['Value'] = np.random.randint(1,100,len(df))
>>> df
Value
Number Letter Color
1 a Red 97
Green 61
b Red 97
Green 98
c Red 91
Green 47
2 a Red 17
Green 63
b Red 26
Green 73
c Red 34
Green 68
但我實際上希望我的索引按“字母、顏色、數字”排序。
我目前這樣做如下:
>>> df.reset_index().set_index(['Letter','Color','Number'])
Value
Letter Color Number
a Red 1 97
Green 1 61
b Red 1 97
Green 1 98
c Red 1 91
Green 1 47
a Red 2 17
Green 2 63
b Red 2 26
Green 2 73
c Red 2 34
Green 2 68
這是最好的方法嗎?
最好使用reorder_levels
來操縱 MultiIndex 級別的順序。 只需按照您想要的順序傳入級別名稱/編號列表:
>>> df.reorder_levels(['Letter','Color','Number'])
Value
Letter Color Number
a Red 1 41
Green 1 56
b Red 1 43
Green 1 42
c Red 1 89
Green 1 18
a Red 2 55
Green 2 93
b Red 2 64
Green 2 9
c Red 2 21
Green 2 93
如果您只想交換兩個級別的位置,也可以使用swaplevel
。
調用MultiIndex.reorder_levels
,然后將新索引分配給您的 DataFrame。
df.index = df.index.reorder_levels(['Letter', 'Color', 'Number'])
df
Value
Letter Color Number
a Red 1 41
Green 1 56
b Red 1 43
Green 1 42
c Red 1 89
Green 1 18
a Red 2 55
Green 2 93
b Red 2 64
Green 2 9
c Red 2 21
Green 2 93
由於 Index 對象是不可變的,因此您無法克服創建新 Index 的困難,但您可以通過調用df.reorder_levels
來避免復制數據。
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.