同じ日付で2行のデータを1行のデータにしたい

Question

pandasのDataFrameで同じ日付で前・後のデータが別の行にある場合で図のように変換したいです．

pivotを使ってマルチカラムにしたりも試しましたがつまりました．

ご教授よろしくお願いします．

元データ

変換後

Answer 1

import pandas as pd
import io

csv_data = '''
day,id,pre_post,a,b,c
2022-01-01 00:00:00,1234,pre,1,2,3
2022-01-01 00:00:00,1234,post,4,5,6
2022-01-15 00:00:00,5678,pre,7,8,9
2022-01-15 00:00:00,5678,post,10,11,12
'''
df = pd.read_csv(io.StringIO(csv_data))

#
cols = ['a', 'b', 'c']
tags = ['pre', 'post']
col_prefix = [{c: f'{t}_' + c for c in cols} for t in tags]
result = (
    df.groupby(['day', 'id'], sort=False)
      .apply(lambda x: pd.concat(
          x.loc[x['pre_post'] == t, cols].iloc[0].rename(col_prefix[i])
          for i, t in enumerate(tags)))
      .reset_index())

print(result)
#
                   day    id  pre_a  pre_b  pre_c  post_a  post_b  post_c
0  2022-01-01 00:00:00  1234      1      2      3       4       5       6
1  2022-01-15 00:00:00  5678      7      8      9      10      11      12

Comments

[cepy]

問題なく動きました．実データはカラムが多いので，こちらのコードを改良して実装していきたいです．ありがとうございます．

Answer 2

以下です。

Comments

[cepy]

ありがとうございます．問題なく動きました．