熊猫：添加新列并按条件从另一个数据帧赋值

Question

我有两个数据帧DF1和DF2

DF1：

id      product
a       a
b       b
c       c
d       d

DF2：

id      documentType      documentUrl
a       3D                https://...a.dxf
a       Image             https://...a.jpg
b       PDF               https://...b.pdf
b       Image             https://...b.jpg
b       Image             https://...b2.jpg
c       PDF               https://...c.pdf

我想在DF1中创建一列“image1”，并根据以下条件赋值。

1. 检查DF1['id']值是否在DF2['id']和DF2['documentType']=='Image'中可用

因此，输出应该如下所示：

id      product      image1
a       a            https://...a.jpg
b       b            https://...b.jpg
c       c            https://...no_image.jpg
d       d            https://...no_image.jpg

不知道如何解决这个问题，但一些想法：

-加入/合并是我的第一个想法，但是如何处理这些条件呢？

-可能使用检查条件的功能映射/应用

DF1['image1'] = DF1['id'].map(DF2.set_index('id')['documentUrl'], condition)

Answer 1

您可以先过滤：

s = (DF2.loc[DF2.documentType=='Image']
        .drop_duplicates('id')
        .set_index('id')['documentUrl']
    )
DF1['image'] = DF1['id'].map(s)

输出：

  id product             image
0  a       a  https://...a.jpg
1  b       b  https://...b.jpg
2  c       c               NaN
3  d       d               NaN