根据来自不同数据框的两列条件将列相乘？

Question

我有两个数据框，如下所示：dfA = Country      City           PopUS           Washington     1000US           Texas          5000CH           Geneva         500CH           Zurich         500dfB = Country      City           Density (pop/km2)US           Washington     10US           Texas          50CH           Geneva         5CH           Zurich         5我想要的是比较列Country和City两个数据帧，以及当它们匹配时，例如：US Washington & US Washington在这两个数据帧中，它都Pop取值并将其除以Density，以便在结果除法中获得一个新area列dfB。第一行结果示例 dfB['area km2'] = 100我试过了，np.where()但它不起作用。关于如何实现这一目标的任何提示？

Answer 1

使用索引匹配和div

match_on = ['Country', 'City']

dfA = dfA.set_index(match_on)

dfA.assign(ratio=dfA.Pop.div(df.set_index(['Country', 'City'])['Density (pop/km2)']))

Country  City      

US       Washington    100.0

         Texas         100.0

CH       Geneva        100.0

         Zurich        100.0

dtype: float64

Answer 2

您还可以使用merge组合两个数据帧并照常划分：

dfMerge = dfA.merge(dfB, on=['Country', 'City'])

dfMerge['area'] = dfMerge['Pop'].div(dfMerge['Density (pop/km2)'])

print(dfMerge)

输出：

  Country        City   Pop  Density (pop/km2)   area

0      US  Washington  1000                 10  100.0

1      US       Texas  5000                 50  100.0

2      CH      Geneva   500                  5  100.0

3      CH      Zurich   500                  5  100.0

Answer 3

你也可以像下面这样使用合并

dfB["Area"] = dfB.merge(dfA, on=["Country", "City"], how="left")["Pop"] / dfB["Density (pop/km2)"]
dfB