pandasで特定の条件の時、その行を横方向にずらしたい

Question

pythonを勉強し始めて１ヶ月ほどの者です。

df = pd.DataFrame([[1,2,3],[4,5,6],[7,8,9]])

例えば、上記のデータフレームで0列目が'4'の時、その行を右にずらしたデータフレームを返したいのですがどうすれば良いでしょうか。下記のようにしたいです。よろしくお願いします。

df = pd.DataFrame([[1,2,3],[nan,4,5],[7,8,9]])

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

エラーメッセージ

該当のソースコード

ソースコード

試したこと

df[df.iloc[:, 0]==4].shift(1, axis=1)

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

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

Answer 1

shift 結果を元の df に設定しましょう。

mask = df.iloc[:, 0] == 4
df.loc[mask] = df.loc[mask].shift(axis=1)

または、mask メソッドを使用します。

df.mask(df.iloc[:, 0] == 4, df.shift(axis=1), inplace=True)

元の df を変更したくない場合、最初の例では、置き換える前に df.copy() を作成する必要があります。2番目の例では、inplace=False を指定して戻り値を使用します。

Comments

[退会済みユーザー]

ありがとうございます、とても参考になりました。