Python プログラミング

Pandasで既存のカラムを元に条件に応じて新しいカラムを追加する【map + lambda】

Kaggleのタイタニックにトライする中で以下のQiitaの記事と出会いました。

Kaggleに登録したら次にやること ~ これだけやれば十分闘える!Titanicの先へ行く入門 10 Kernel ~

 

この中で、「FamilySize」という乗船客の家族数を新しい特徴量として追加するのですが、

その後に家族数が1人かそれ以外かという「IsAlone」という特徴量を家族数を元にさらに追加します。

 

この時に、良い感じにカラム追加を1行を追加できないかと言う事で、Pandas超初心者なりに調べたところ

以下のような書き方でいけました。

 

 

mapメソッドは各カラムに引数で渡した関数の処理を行い、新しいカラム(Series)を返却します。

勿論元のカラムには変更は加わりません。また、fillnaなどにある「in_place」の指定もないとのこと。

 

pandasすごいですね。

 

-Python, プログラミング
-, , ,

© 2020 クラインの備忘壺