pandasの使い方

1 はじめに
2 サンプルコード
3 さいごに

はじめに

こんにちは。はんぺんです。

pandasの便利な使い方を見つけたのでメモがてら記事にしようと思います。

サンプルコード

import

import pandas as pd

1	import pandas as pd

データフレームの生成

df = pd.DataFrame(['a', 'b', 'c', 'a', 'a', 'b', 'a', 'c'], columns=['alphabet'])

1	df = pd.DataFrame(['a', 'b', 'c', 'a', 'a', 'b', 'a', 'c'], columns=['alphabet'])

このようにデータフレームが生成されます

フラグカラム生成

df['flag'] = 0

1	df['flag'] = 0

フラグ用のカラムを生成します

このようにフラグのカラムが追加されます。

フラグを割り当てる

flag_dict = {'a':1, 'b':2, 'c':3}

1	flag_dict = {'a':1, 'b':2, 'c':3}

aを1, bを2,cを3に割り当てます

df.loc[df['alphabet']=='a', 'flag'] = flag_dict['a']

1	df.loc[df['alphabet']=='a', 'flag'] = flag_dict['a']

このようにかけば、aのフラグカラムに1が入ります。

確認してみると。

aのところだけ1に変わっています。

同様にb,cも書いてみると。

df.loc[df['alphabet']=='b', 'flag'] = flag_dict['b']
df.loc[df['alphabet']=='c', 'flag'] = flag_dict['c']

1 2	df.loc[df['alphabet']=='b', 'flag'] = flag_dict['b'] df.loc[df['alphabet']=='c', 'flag'] = flag_dict['c']

このようにフラグに値を入れることができました！

コードまとめ

以下が全コードになります。

import pandas as pd

df = pd.DataFrame(['a', 'b', 'c', 'a', 'a', 'b', 'a', 'c'], columns=['alphabet'])
df['flag'] = 0
flag_dict = {'a':1, 'b':2, 'c':3}
df.loc[df['alphabet']=='a', 'flag'] = flag_dict['a']
df.loc[df['alphabet']=='b', 'flag'] = flag_dict['b']
df.loc[df['alphabet']=='c', 'flag'] = flag_dict['c']