データフレームの置換

Question

日付ごとにカラムBとCを上下反転させたい

date	A	B	C
2020-08-02	10	20	30
2020-08-02	10	10	10
2020-08-03	20	20	10
2020-08-03	50	50	50
...
...
2020-09-02	20	20	20
2020-09-02	40	40	40

変更後

date	A	B	C
2020-08-02	10	10	10
2020-08-02	10	20	30
2020-08-03	20	50	50
2020-08-03	50	20	10
...
...
2020-09-02	20	40	40
2020-09-02	40	20	20

date
out:
DatetimeIndex(['2020-08-02', '2020-08-03', '2020-08-04', '2020-08-05',
......
'2020-08-30', '2020-08-31', '2020-09-01', '2020-09-02'],

for i in date:
    df.loc[df['date'] == i, ['B','C']] = df[df['date'] == i ][['B','C']][::-1]

上記のコードで試みましたが置換ができません。
どなたかご教授いただけないでしょうか？

Comments

[Daregada]

同じ日付のデータが、かならず2行ペアになっていて、それぞれのペアでB列のデータを交換、C列のデータを交換したい、ということでいいのでしょうか。

Answer 1

for文ではなく、.groupby()メソッドを用いて'B', 'C'列を反転させます。例えば、

df[['B', 'C']] = df.groupby('date')['B', 'C'].transform(lambda s: s[::-1])

もし'date'列がソート済み、かつ全てが2行ずつのグループになっている場合は以下のようにインデックスを用いて反転させるのが容易です。

df['B'][1::2], df['B'][::2] = df['B'][::2].to_list(), df['B'][1::2].to_list()
df['C'][1::2], df['C'][::2] = df['C'][::2].to_list(), df['C'][1::2].to_list()

df[['B', 'C']] = df[['B', 'C']].to_numpy().reshape(len(df)//2, 2, 2)[:,::-1,:].reshape(len(df),-1)

df[['B', 'C']] = df[['B', 'C']].iloc[np.arange(len(df)).reshape(len(df)//2, 2)[:,::-1].ravel()]

Comments

[hachiroku]

いくつか例をあげていただいて大変ありがたいです。ありがとうございます。

Answer 2

泥臭く、2行読むごとにB列の内容を入れ替え、C列の内容を入れ替えるコード。

import pandas as pd
import io
import sys

txt = """
date,A,B,C
2020-08-02,10,20,30
2020-08-02,10,10,10
2020-08-03,20,20,10
2020-08-03,50,50,50
2020-09-02,20,20,20
2020-09-02,40,40,40
"""

df = pd.read_table(io.StringIO(txt), sep=",", parse_dates=['date'])

row1 = None
for row2 in df.itertuples():
    if row1 is None:
        row1 = row2
        continue
    if row1.date != row2.date:
        sys.exit(1)
    #print(row1, row2, sep='\n', end='\n' + '-' * 8 + '\n')
    df.at[row1.Index, 'B'] = row2.B
    df.at[row1.Index, 'C'] = row2.C
    df.at[row2.Index, 'B'] = row1.B
    df.at[row2.Index, 'C'] = row1.C
    row1 = None

df.set_index('date', inplace=True)
print(df)

             A   B   C
date
2020-08-02  10  10  10
2020-08-02  10  20  30
2020-08-03  20  50  50
2020-08-03  50  20  10
2020-09-02  20  40  40
2020-09-02  40  20  20

Comments

[hachiroku]

ありがとうございます。参考にさせていただきます。