比较python中的两列并在第一个表中返回匹配项

yh2wf1be 于 2021-09-08 发布在 Java

关注(0)|答案(1)|浏览(421)

帮助按周期比较不同表中的两列，并将匹配项返回到第一个表。

data1:
|name   | revenue |
|-------|---------|
|Alice  | 700     |
|Bob    | 1000    |
|Gerry  | 300     |
|Alex   | 600     |
|Kyle   | 800     |

data2:
|Name   | revenue |
|-------|---------|
|Bob    | 900     |
|Gerry  | 400     |

result data1:
|name   | revenue  |  name_result |
|-------|----------|--------------|
|Alice  | 700      |              |
|Bob    | 1000     |  Bob         |
|Gerry  | 300      |  Gerry       |
|Alex   | 600      |              |
|Kyle   | 800      |              |

我尝试使用此代码，但得到了所有空值：

import pandas as pd
import numpy as np

def group_category(category):
    for name in data['name']: 
        if name in data2['Name']:
            return name
        else: name = ''
        return name 
data['name_result'] = data['name'].apply(group_category)

python compare cycle

来源：https://stackoverflow.com/questions/68316518/compare-two-column-in-python-and-return-matches-in-first-table

1条答案

按热度按时间

t5zmwmid1#

使用：

def group_category(category):
    if category in df2['Name'].unique():
            return category
    else:
        return ''

# Finally:

# Since you are going to use this function on Series so used map() in place of apply()

df1['name_result']=df1['name'].map(group_category)

或
通过 isin() 及 where() :

df1['name_result']=df1['name'].where(df1['name'].isin(df2['Name']),'')

赞(0）回复(0）举报 2021-09-08

我来回答

比较python中的两列并在第一个表中返回匹配项

1条答案

相关问题

热门标签

最新问答