Q&A
df2に列を追加しようとした場合、カテゴリ大がファッションじゃないときはOKとNGをどのように判定するのか指示しないといけないのではないでしょうか。
データフレームについて、カテゴリ別に基準値を設けて、基準値以下なら"OK"、基準値を超えるなら"NG"を列に追加したです。
以下のようなコードで追加しようとしましたが、NaNとなります。
列を追加する方法をご教授いただけないでしょうか。
df2['結果'] = df2.loc['ファッション']['金額'].apply(lambda x : 'OK' if x <= 30000 else 'NG')
今回以下のような処理を行なっています。
①以下のようなCSVデータを読み込んでデータフレームを作成しています。
df = pd.read_csv('/input.csv', encoding = "shift-jis") df['日付'] = pd.to_datetime(df['日付'],format='%Y/%m/%d')
dfの中身
||項目|金額|カテゴリ大|カテゴリ小|
|:--|:--:|--:|
|日付||||
|2019-11-02|ユニクロ|10000|ファッション|NaN
|2019-11-16|電気|12000|固定費|光熱費
|・・・|・・・|・・・|・・・|・・・
|2021-03-10|スーパー|5000|食費|食材費
②上記のdfについて、カテゴリ大毎に月の合計金額を算出しています。
df2 = df.groupby(['カテゴリ大']).resample('M').sum()
df2の中身
| 金額 | ||
|---|---|---|
| カテゴリ大 | 日付 | |
| ファッション | 2019-11-30 | 11000 |
| 2019-12-31 | 40000 | |
| 2020-01-31 | 20000 | |
| ・・・ | ・・・ | ・・・ |
| 食費 | 2019-11-30 | 30000 |
| 2019-12-31 | 50000 | |
| 2020-01-31 | 45000 |
上記②のdfにファッションなら、30000円以下なら"OK"、それ以外なら"NG"を列に追加したいのですが、NaNとなります。
df2['結果'] = df2.loc['ファッション']['金額'].apply(lambda x : 'OK' if x <= 30000 else 'NG')
回答2件
0
ベストアンサー
欲しいのはこういうものでしょうか。
python
1>>> df_limit = pd.DataFrame({'カテゴリ大':['ファッション', '固定費', '食費'], '予算':[30000, 50000, 40000]}) 2>>> print(df) 3 項目 金額 カテゴリ大 カテゴリ小 4日付 52019-11-02 ユニクロ 10000 ファッション NaN 62019-11-03 丸井 25000 ファッション NaN 72019-11-16 電気 12000 固定費 光熱費 82019-11-15 スーパー 8000 食費 食材費 92019-12-01 スーパー 5000 食費 食材費 10>>> df2 = df.groupby(['カテゴリ大']).resample('M').sum().reset_index() 11>>> print(df2) 12 カテゴリ大 日付 金額 130 ファッション 2019-11-30 35000 141 固定費 2019-11-30 12000 152 食費 2019-11-30 8000 163 食費 2019-12-31 5000 17>>> df3 = pd.merge(df2, df_limit, on='カテゴリ大').set_index('日付') 18>>> df3['結果'] = df3.apply(lambda row: 'OK' if row['金額'] <= row['予算'] else 'NG', axis=1) 19>>> df3.drop(['予算'], axis=1, inplace=True) 20>>> print(df3) 21 カテゴリ大 金額 結果 22日付 232019-11-30 ファッション 35000 NG 242019-11-30 固定費 12000 OK 252019-11-30 食費 8000 OK 262019-12-31 食費 5000 OK
投稿2021/03/13 09:10
総合スコア24670
ご回答ありがとうございます。
mergeメソッド知らなかったです。
勉強になります。
ご教授いただいた形式の他に、私が記載しましたdf2にそのまま結果列を追加する方法はありますでしょうか？
こんな感じのdfにしたいです。
金額 結果
カテゴリ大 日付
ファッション 2019-11-30 10000 OK
2019-12-31 40000 NG
2020-01-31 20000 OK
・・・ ・・・ ・・・
食費 2019-11-30 30000 OK
2019-12-31 50000 NG
2020-01-31 45000 NG
あの後に、
df4 = df3.reset_index().set_index(['カテゴリ大', '日付'])
を行ってください。
上記回答いただいた内容で期待する結果になりました！
ありがとうございます。
0
Python
1df2.loc['ファッション','結果']=list(df2.loc['ファッション','金額'].apply(lambda x : 'OK' if x<=30000 else 'NG'))
とか
投稿2021/03/13 13:59
総合スコア282
あなたの回答
tips
プレビュー
