さわやかに

Pythonのことだったり子供のことだったり

Python Pandas DataFrame 便利なApply

import pandas as pd
df_fruit = pd.DataFrame({'フルーツ': ['りんご', 'ばなな', 'ぶどう', 'みかん', 'もも', 'なし','すいか', 'メロン', 'パイナップル', 'かき'],
                        '価格': [50, 100, 500, 30, 400, 200, 1000, 2000, 300, 80]})
f:id:kinakobanana:20201002232505p:plain
df_fruit

価格をもとに新たな価格指数カラムを作る
条件は500円以下をLow、500円を超え1000円以下をMid、1000円を超えるものをHighとする

df judge(ex):
    if ex <= 500:
        return 'Low'
    elif 500 < ex <= 1000:
        return 'Mid'
    else:
        return 'High'
df_fruit.loc[:, '価格指数'] = df_fruit.loc[:, '価格'].apply(judge)
f:id:kinakobanana:20201002235205p:plain
df_fruit

できました



価格指数に入力されている['Low', 'Mid'. 'High']のデータを、3カラムに分類し該当箇所には1を、非該当箇所には0を入れたDataFrameを作成する

df_price = pd.get_dummies(df_fruit.loc[:, '価格指数'], prefix='価格')
f:id:kinakobanana:20201003000005p:plain
df_price

できました

Pythonによるあたらしいデータ分析の教科書」を参考にさせてもらいました。