Python Pandas DataFrame 便利なApply
import pandas as pd df_fruit = pd.DataFrame({'フルーツ': ['りんご', 'ばなな', 'ぶどう', 'みかん', 'もも', 'なし','すいか', 'メロン', 'パイナップル', 'かき'], '価格': [50, 100, 500, 30, 400, 200, 1000, 2000, 300, 80]})
価格をもとに新たな価格指数カラムを作る
条件は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)
できました
価格指数に入力されている['Low', 'Mid'. 'High']のデータを、3カラムに分類し該当箇所には1を、非該当箇所には0を入れたDataFrameを作成する
df_price = pd.get_dummies(df_fruit.loc[:, '価格指数'], prefix='価格')
できました
「Pythonによるあたらしいデータ分析の教科書」を参考にさせてもらいました。