追加一個級別（具有固定值）到大熊貓系列/數據幀

Question

我與多指數大熊貓系列如下：

import pandas as pd 
import numpy as np 
idx = pd.MultiIndex.from_product([['A','C'],range(5)], names=['category_1','number']) 
np.random.seed(0) 
s = pd.Series(index=idx, data = np.random.randn(len(idx))) 

I：

category_1 number 
A   0   1.764052 
      1   0.400157 
      2   0.978738 
      3   2.240893 
      4   1.867558 
C   0  -0.977278 
      1   0.950088 
      2  -0.151357 
      3  -0.103219 
      4   0.410599 

它是從這個代碼生成想添加另一個級別，稱爲category_2以具有固定值的索引（即D）得到以下結果：

category_1 category_2 number 
A   D   0   1.764052 
         1   0.400157 
         2   0.978738 
         3   2.240893 
         4   1.867558 
C   D   0  -0.977278 
         1   0.950088 
         2  -0.151357 
         3  -0.103219 
         4   0.410599 

我一直用這個哈克的方式做到這一點：

df =s.to_frame('dummy') 
df['category_2'] = 'D' 
df.set_index('category_2', append = True, inplace = True) 
df = df.reorder_levels([0,2,1]) 
res = df['dummy'] 

有沒有更好的（更簡潔/ Python化）的方式來增加固定值來對大熊貓系列/數據框中現有水平的水平？

Answer 1

您需要創建新MultiIndex，然後替換舊的：

#change multiindex 
new_index = list(zip(s.index.get_level_values('category_1'), 
        ['D'] * len(s.index), 
        s.index.get_level_values('number'))) 
print (new_index) 
[('A', 'D', 0), ('A', 'D', 1), 
('A', 'D', 2), ('A', 'D', 3), 
('A', 'D', 4), ('C', 'D', 0), 
('C', 'D', 1), ('C', 'D', 2), 
('C', 'D', 3), ('C', 'D', 4)] 
s.index = pd.MultiIndex.from_tuples(new_index, 
            names=['category_1','category_2','number']) 
print (s) 
category_1 category_2 number 
A   D   0   1.764052 
         1   0.400157 
         2   0.978738 
         3   2.240893 
         4   1.867558 
C   D   0  -0.977278 
         1   0.950088 
         2  -0.151357 
         3  -0.103219 
         4   0.410599 
dtype: float64 

與MultiIndex.from_product另一種很好的解決方案 - 有點改變comment：

s.index = pd.MultiIndex.from_product([s.index.levels[0], 
             ['D'], 
             s.index.levels[1]], names= ['c1','c2','number']) 
print (s) 
c1 c2 number 
A D 0   1.764052 
     1   0.400157 
     2   0.978738 
     3   2.240893 
     4   1.867558 
C D 0  -0.977278 
     1   0.950088 
     2  -0.151357 
     3  -0.103219 
     4   0.410599 
dtype: float64 

或者：

s.index = pd.MultiIndex.from_product([s.index.get_level_values('category_1').unique(), 
             ['D'], 
             s.index.get_level_values('number').unique()], 
            names= ['c1','c2','number']) 
print (s) 
c1 c2 number 
A D 0   1.764052 
     1   0.400157 
     2   0.978738 
     3   2.240893 
     4   1.867558 
C D 0  -0.977278 
     1   0.950088 
     2  -0.151357 
     3  -0.103219 
     4   0.410599 
dtype: float64