DataFrame型で値を指定しているのに代入できないときの対処法

データフレームで場所を指定しているのに反映されない！？なぜ！！？？

pandasの深い深い闇にはまり２時間以上溶かしたので、共有をします。

Contents

やりたいこと
- 沼にはまるパターン
- きちんと代入できるパターン
まとめ

やりたいこと

DataFrame型をの変数に、indexとcolumnsを指定して、それと同じshapeのnumpy配列を代入する。

沼にはまるパターン

# きれいにデータフレーム型を表示させるための呪文
# pyファイルに変換をしても怒られないのでオススメです。
from IPython.core.interactiveshell import InteractiveShell
InteractiveShell.ast_node_interactivity = "all"

a = np.array([[1,2,3],[2,3,4],[3,4,5],[1,2,3]])
b = np.array([5,5,5,5])
df = pd.DataFrame(a)
df
idx = [False, True, False, True ]
cols = [1,2]
df[idx][cols]
df[idx][cols] = [100,100]
df

df[idx][cols]の部分で、参照したい部分は確かに参照できていることが分かります。

なのに、なんといいうことでしょう！

代入したはずの[100, 100]が反映されていないではないですか！！

そうです。これが今回お伝えしたかった沼です。

df[idx][cols]という記法では参照しかされないらしいです。

おぞましすぎます。

代入されないくせに、何もエラーを吐かないのです。

悪魔です。

では、どうしたらきちんと代入をしてくれるのか？？

きちんと代入できるパターン

from IPython.core.interactiveshell import InteractiveShell
InteractiveShell.ast_node_interactivity = "all"

a = np.array([[1,2,3],[2,3,4],[3,4,5],[1,2,3]])
b = np.array([5,5,5,5])
df = pd.DataFrame(a)
df
idx = [False, True, False, True ]
cols = [1,2]
df.loc[idx,cols]
# ココをlocに変更！
df.loc[idx,cols] = [100,100]
df