質問しようと思ってあちこちいじってたら上手く行ったので、備忘録がてら登録。
主題
表1のデータから表2を作りたい。
表1 生データ
| 総額 | 食費 | 光熱費 | 水道費 |
|---|---|---|---|
| 2000 | 2000 | 0 | 0 |
| 1000 | 0 | 1000 | 0 |
| 1500 | 0 | 0 | 1500 |
表2 作りたいデータ
| 総額 | 食費 | 光熱費 | 水道費 | 分類 |
|---|---|---|---|---|
| 2000 | 2000 | 0 | 0 | 食費 |
| 1000 | 0 | 1000 | 0 | 光熱費 |
| 1500 | 0 | 0 | 1500 | 水道費 |
食費・光熱費・水道費はどれか一つに値が入るようになっている
食費1000＋水道費1000＝総額2000、というようなことはない。
自分で試したこと
import pandas as pd
df = pd.DataFrame(
data=[{'総額': 2000, '食費': 2000,'光熱費': 0,'水道費': 0,},
{'総額': 1000, '食費': 0,'光熱費': 1000,'水道費': 0,},
{'総額': 1500, '食費': 0,'光熱費': 0,'水道費': 1500,}])
df_temp= df.drop("総額",axis=1)
df_temp["分類"]=df_temp.idxmax(axis=1)
df=df.merge(df_temp)
今回は3列の中の最大値が欲しかったので、idxmaxで一発で列ラベルを拾えた。
idxmax、idxminが使えるならこれが良さそう