コードと日付が変わった場合、フラグを立てたい

Question

前提・実現したいこと

EDINETコードと会計期間終了日のいずれかが変わった場合、その直前にフラグ１、変わらない場合にその直前にフラグ０を立てたい。

発生している問題・エラーメッセージ

＜ソースコード実行結果＞
EDINETコード 会計期間終了日
E00004 2016-05-31 0.0
2019-05-31 1.0
E00008 2016-03-31 0.0
2016-03-31 0.0
2020-03-31 1.0
...
E35303 2020-03-31 1.0
E35307 2019-12-31 1.0
E35323 2019-12-31 1.0
E35335 2020-03-31 1.0
E35487 2020-03-31 1.0

該当のソースコード

y_column = 'フラグ'
flag = np.array([np.nan] * len(df), dtype=np.float32)
edinet_codes = df.index.map(lambda x: x[0])
df['会計期間終了日'] = df.index.map(lambda x: x[0])

prev_code = None
prev_date = None

for i, [code, date] in enumerate(zip(edinet_codes, df['会計期間終了日'])):
if (prev_date != date) | (prev_code !=code):
flag[i -1] = 1
prev_date = date
prev_code = code
else:
flag[i -1] =0

df[y_column] = flag
print(df[y_column])

試したこと

ここに問題に対して試したことを記載してください。

補足情報（FW/ツールのバージョンなど）

ここにより詳細な情報を記載してください。

Comments

[can110]

タグに「python」と「pandas」を加えると、より多くの人の目にとまりやすくなります。
また、質問中の「その直前」というのがよく分からないので、得たい結果の例を示してください。

[YT1967]

teratail初心者なので、アドバイス助かります。
直線の表現分かりにくくすみません。
下記回答いただいた2番目の例が取得したい結果でした。
ありがとうございました。

Answer 1

欲しい結果が明確に理解できていませんが、.shiftを使って前後の行との値を比較することで結果が得られるかと思います。

import pandas as pd
import numpy as np
from io import StringIO

s = """code,date
E1,2021-06-01
E1,2021-06-01
E2,2021-06-01
E2,2021-06-01
E2,2021-06-02
E2,2021-06-02
E3,2021-06-03
E4,2021-06-04"""
df = pd.read_csv(StringIO(s), parse_dates=['date'])

# 変化した行が１になる
# 先頭行は無条件で１になる
df_prev = df.shift(1)
ret = (df['code'] != df_prev['code']) | (df['date'] != df_prev['date'])
df['flag'] = ret.astype(int)
print(df)
"""
  code       date  flag
0   E1 2021-06-01     1
1   E1 2021-06-01     0
2   E2 2021-06-01     1
3   E2 2021-06-01     0
4   E2 2021-06-02     1
5   E2 2021-06-02     0
6   E3 2021-06-03     1
7   E4 2021-06-04     1
"""

# 変化する直前の行が１になる
# 最終行は無条件で１になる
df_next = df.shift(-1)
ret = (df['code'] != df_next['code']) | (df['date'] != df_next['date'])
df['flag'] = ret.astype(int)
print(df)
"""
  code       date  flag
0   E1 2021-06-01     0
1   E1 2021-06-01     1
2   E2 2021-06-01     0
3   E2 2021-06-01     1
4   E2 2021-06-02     0
5   E2 2021-06-02     1
6   E3 2021-06-03     1
7   E4 2021-06-04     1
"""

Comments

[YT1967]

.shiftを使って比較すればよいとのこと、大変勉強になりました。
ありがとうございました。