gpt4 book ai didi

python - 使用内部连接连接两个数据框

转载 作者:行者123 更新时间:2023-11-28 20:30:55 24 4
gpt4 key购买 nike

假设我有这两个数据集:

Data frame 1:
X | date
a | 1/1/2018
a | 1/2/2018
...
b | 1/1/2018

和 df 2:

Holiday
1/1/2018
5/1/2018

在第一个数据集中添加一个新列的优雅方法是什么,当日期与第二个数据集匹配时为 1?

非常感谢!

最佳答案

这是通过 isin() 方法实现的。

df1 = pd.DataFrame({'date': ['1/1/2018', '1/2/2018', '1/1/2018']})
df2 = pd.DataFrame({'Holiday': ['1/1/2018', '5/1/2018']})
df1
# date
# 0 1/1/2018
# 1 1/2/2018
# 2 1/1/2018
df2
# Holiday
# 0 1/1/2018
# 1 5/1/2018

df1['is_holiday'] = df1.date.isin(df2.Holiday).astype(int)

df1
# date is_holiday
# 0 1/1/2018 1
# 1 1/2/2018 0
# 2 1/1/2018 1

关于python - 使用内部连接连接两个数据框,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/56318192/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com