2つのpandasデータフレームを使った置換をしたい場合

前提・実現したいこと

pandasで、以下のdf1とdf2があります。
df2にデータが存在する場合、df1のフラグを1に変更したいです。
（日付、時間、IDを条件にしたいです。）

df1
||日付|時間|ID|名前|金額|フラグ|
|:--|:--:|--:|
|1|20210101|00:00|0001|hanako|1000|0|
|2|20210102|00:00|0002|taro|1000|0|
|3|20210102|13:00|0002|taro|500|1|
|4|20210103|01:00|0003|mami|300|1|
|5|20210104|10:00|0004|aki|5000|0|
|6|20210105|00:30|0005|huyu|7000|0|

df2
||日付|時間|ID|名前|
|:--|:--:|--:|
|1|20210102|00:00|0002|taro
|2|20210105|00:30|0005|huyu

実現したいこと
||日付|時間|ID|名前|金額|フラグ|
|:--|:--:|--:|
|1|20210101|00:00|0001|hanako|1000|0|
|2|20210102|00:00|0002|taro|1000|1|
|3|20210102|13:00|0002|taro|500|1|
|4|20210103|01:00|0003|mami|300|1|
|5|20210104|10:00|0004|aki|5000|0|
|6|20210105|00:30|0005|huyu|7000|1|

行動規範の内容に同意します

回答1件

ベストアンサー

2つのデータフレームの日付・時間・IDをインデックスに指定してから、df1.loc[df2.index, 'フラグ'] = 1すればいいのでは。

Python
1import pandas as pd
2import io
3
4txt1 = """
5日付,時間,ID,名前,金額,フラグ
620210101,00:00,0001,hanako,1000,0
720210102,00:00,0002,taro,1000,0
820210102,13:00,0002,taro,500,1
920210103,01:00,0003,mami,300,1
1020210104,10:00,0004,aki,5000,0
1120210105,00:30,0005,huyu,7000,0
12"""
13txt2 = """
14日付,時間,ID,名前
1520210102,00:00,0002,taro
1620210105,00:30,0005,huyu
17"""
18
19# 日本語ヘッダーのずれを防ぐ
20pd.set_option('display.unicode.east_asian_width', True)
21
22df1 = pd.read_csv(io.StringIO(txt1), dtype={'ID': str})
23df2 = pd.read_csv(io.StringIO(txt2), dtype={'ID': str})
24
25df1.set_index(['日付', '時間', 'ID'], inplace=True)
26df2.set_index(['日付', '時間', 'ID'], inplace=True)
27
28df1.loc[df2.index, 'フラグ'] = 1
29
30df1.reset_index(inplace=True)
31df2.reset_index(inplace=True)
32
33print(df1)
34

result
1       日付   時間    ID    名前  金額  フラグ
20  20210101  00:00  0001  hanako  1000       0
31  20210102  00:00  0002    taro  1000       1
42  20210102  13:00  0002    taro   500       1
53  20210103  01:00  0003    mami   300       1
64  20210104  10:00  0004     aki  5000       0
75  20210105  00:30  0005    huyu  7000       1

投稿2021/01/20 17:14

編集2021/01/21 00:32