下面是一个示例代码,比较两个表并将Table A的行ID写入Table B:
import pandas as pd
# 创建示例数据
data_a = {'ID': [1, 2, 3, 4, 5],
'Name': ['Alice', 'Bob', 'Charlie', 'David', 'Emma']}
data_b = {'ID': [2, 4, 6, 8, 10],
'Age': [25, 30, 35, 40, 45]}
table_a = pd.DataFrame(data_a)
table_b = pd.DataFrame(data_b)
# 比较两个表并将Table A的行ID写入Table B
table_b['A_ID'] = table_b['ID'].apply(lambda x: x in table_a['ID'].values)
print(table_b)
运行上述代码,输出结果如下:
ID Age A_ID
0 2 25 True
1 4 30 True
2 6 35 False
3 8 40 False
4 10 45 False
在上述代码中,我们使用pandas库创建了两个表table_a
和table_b
。然后,我们使用apply
函数和lambda
表达式将Table A的行ID与Table B的ID进行比较,并将结果写入Table B的新列A_ID
中。如果ID在Table A中存在,则对应的A_ID
值为True,否则为False。