多索引熊猫数据框和.diff()

Sir*_*irC 1 python multi-index pandas

假设我有以下多索引熊猫数据框:

                     A    B
Date        Code     
01-01-2017  s1       1    2
            s2       3    1
01-02-2017  s1       2    2    
            s2       3    3
Run Code Online (Sandbox Code Playgroud)

这里讨论的内容不同,我不希望.diff(1)对具有相同日期的值进行操作并在每个新日期进行重置,我希望.diff(1)提供以下输出:

                     A    B
Date        Code     
01-01-2017  s1       Nan  Nan
            s2       Nan  Nan
01-02-2017  s1       1    0    
            s2       2    1
Run Code Online (Sandbox Code Playgroud)

也就是说,.diff(1)通过级别0的索引的差异值指定的块进行差异,而不是像建议的链接那样在块内进行差异。换句话说,我想通过条目减去这两个矩阵:

|2    2|  _  |1    2| 
|3    3|     |3    1|
Run Code Online (Sandbox Code Playgroud)

jez*_*ael 6

您可以groupby在第二级使用DataFrameGroupBy.diff

df = df.sort_index()

df = df.groupby(level=1).diff()
print (df)
                   A    B
Date       Code          
01-01-2017 s1    NaN  NaN
           s2    NaN  NaN
01-02-2017 s1    1.0  0.0
           s2    0.0  2.0
Run Code Online (Sandbox Code Playgroud)

因为如果使用第一级:

df = df.groupby(level=0).diff()
print (df)
                   A    B
Date       Code          
01-01-2017 s1    NaN  NaN
           s2    2.0 -1.0
01-02-2017 s1    NaN  NaN
           s2    1.0  1.0
Run Code Online (Sandbox Code Playgroud)