使用一个数据框中的一个元素来查找另一个数据框中的所有信息？ Python

Question

通过运行 2 个 sql 查询，我在 python 中有 2 个数据框。 表A：

Name	Email	Code
A	E	E1,E2,E3
B	F	F1,F2
C	G	G1
...	...	...

表 B：

Info1	Code	Info2
...	E1	...
...	F1	...
...	E2	...
...	E3	...
...	G1	...
...	F2	...

因为它们是 Python 中的数据框，所以我想一一阅读电子邮件，并在 TableA 中找到它的代码列表/数组。 并使用列表/数组中的每个元素查找表 B 中的所有数据，创建 Excel 并发送到信息主机。 例如， 我想得到下面的excel结果并将其发送给E。

Info1	Code	Info2
...	E1	...
...	E2	...
...	E3	...

例如， 我想得到下面的excel结果并将其发送给F。

Info1	Code	Info2
...	F1	...
...	F2	...

例如， 我想得到下面的excel结果并将其发送给G。

Info1	Code	Info2
...	G1	...

请在 Python 中使用数据框。 非常感谢

Answer 1

你可以试试这样的：

new_df = df1.assign(Code=df['Code'].str.split(',')).explode('Code').merge(df2)
for email, sub_df in new_df.groupby(['Email']):
    print(f'Sending df to {email}...')
    print(sub_df.to_csv())

输出：

Sending df to E...
,Name,Email,Code,Info1,Info2
0,A,E,E1,...,...
1,A,E,E2,...,...
2,A,E,E3,...,...

Sending df to F...
,Name,Email,Code,Info1,Info2
3,B,F,F1,...,...
4,B,F,F2,...,...

Sending df to G...
,Name,Email,Code,Info1,Info2
5,C,G,G1,...,...