条件式に合致する数値を代入したい

Question

pandasのある列の条件に合致するとき、違う列に番号を代入したいのですがうまくできません。

データセット（df）

index	A	B
1	20	0
2	40	0
3	60	0
4	80	0
5	100	0

やりたいこと

dfのA列で条件式に合致するとき、B列に数値を代入したいです。
（条件式）A列　　B列
・0 < i <= 20　　 1
・21 < i <= 40 2
・41 < i <= 60 3
・61 < i <= 80 4
・81 < i <= 100 5

やってみたこと

for i in df['A']:
    if 0 < i <= 20 : df['B']=1
    elif 21 < i <= 40 : df['B']=2
    elif 41 < i <= 60 : df['B']=3
    elif 61 < i <= 80 : df['B']=4
    elif 81 < i <= 100 : df['B']=5
    else: nothing

df['B']のoutputは全て5になります。df['B']をprintにすると、1,2,3,4,5となりますのでdf['B]への代入が理解できていないと思います。
また、可能なら、20刻みの一定間隔なので一括で処理する方法がありましたら、ご教示頂けると幸甚です。

よろしくお願いします。

Comments

[meg_]

df['A']の最終行でelif 81 < i <= 100 : df['B']=5がTrueとなりdf['B']=5の結果が残るからです。理解できなければ、上記コードでiとdfの値を出力しながら確認してみてください。
やりたいことについては「pandas 条件」で検索すれば情報が見つかります。

Answer 1

df['B']=1はB列全体への代入です。よって毎回全行書き換わってます。
やり方は色々ありますので一例として

import pandas as pd
df=pd.DataFrame({"A":[20,40,60,80,100],"B":[0,0,0,0,0]})
for i in range(len(df)):
    if 0 < df.iat[i,0] <= 20 : df.iat[i,1]=1
    elif 21 < df.iat[i,0] <= 40 : df.iat[i,1]=2
    elif 41 < df.iat[i,0] <= 60 : df.iat[i,1]=3
    elif 61 < df.iat[i,0] <= 80 : df.iat[i,1]=4
    elif 81 < df.iat[i,0] <= 100 : df.iat[i,1]=5
    else: nothing

Comments

[hidemomo]

ありがとうございます。iatを使うのですね。助かりました。