时间戳

首页 > 解决方案 > 根据两个索引减去两个多索引数据帧

python - 根据两个索引减去两个多索引数据帧

问题描述

我有两个数据框，如下所示：

df1 = 
          val
id1 id2
foo  1     50
     2    100
     4    200
bar  1     25
     4    100
     8    200
baz  2     75
     4    150


df2 = 
          val
id1 id2
foo  1     60
     2    200
     3    100
bar  1     50
     4    101
     8    202
baz  2     70
     5    155

我正在尝试创建一个新的数据框，它比较两个索引并val仅在两个索引匹配时减去该列。

如何生成数据：

import pandas as pd
d1 = {'id1':['foo','foo', 'foo', 'bar', 'bar', 'bar','baz', 'baz'],
   'id2':[1, 2, 4, 1, 4, 8, 2, 4], 
   'val':[50, 100, 200, 25, 100, 200, 75, 150]}
df1 = pd.DataFrame(data=d1)
df1.set_index(['id1', 'id2'], inplace=True)

 

import pandas as pd
d2 = {'id1':['foo','foo', 'foo', 'bar', 'bar', 'bar','baz', 'baz'],
   'id2':[1, 2, 3, 1, 4, 8, 2, 5], 
   'val':[60, 200, 100, 50, 101, 202, 70, 155]}
df2 = pd.DataFrame(data=d2)
df2.set_index(['id1', 'id2'], inplace=True)

我对 python 很陌生，多索引是我现在生活的祸根。

标签： pythonpandasmultidimensional-array

解决方案

您可以直接减去两个数据帧。它将减去索引匹配的值。

如下所示，

res=df1-df2

输出：

           val
id1 id2       
bar 1    -25.0
    4     -1.0
    8     -2.0
baz 2      5.0
    4      NaN
    5      NaN
foo 1    -10.0
    2   -100.0
    3      NaN
    4      NaN
           val

如果您只想查看匹配的记录，请尝试如下所示，

res[res['val'].notnull()]

输出：

id1 id2       
bar 1    -25.0
    4     -1.0
    8     -2.0
baz 2      5.0
foo 1    -10.0
    2   -100.0

有关更多详细信息，请点击此链接多索引

如果您有多个列并且想要在下面使用单列执行减法

df1['res']=df1['val']-df2['val']

输出：

         val    res
id1 id2            
foo 1     50  -10.0
    2    100 -100.0
    4    200    NaN
bar 1     25  -25.0
    4    100   -1.0
    8    200   -2.0
baz 2     75    5.0
    4    150    NaN

上面的代码片段对列执行减法val并将其存储到名为的新列中res

推荐阅读