Question

我有两个看起来像下面的数据框：

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的新手，多索引是我一生的祸根。

Answer 1

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

如下所示，

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

有关更多详细信息，请点击此链接 multi-indexing

如果您有多列，并且想对单列进行减法，请在下面使用

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的新列中

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

1 个答案: