pandasライブラリでif条件の適用
2021-03-30
2021-03-30
前置き
pandasライブラリではif条件を適用する方法がいくつかあります。ここではまとめていきます。
Lambdaを使って処理する
```import pandas as pd
data = {'title': ['t1','t2','t1','t1','t2']}
df = pd.DataFrame(data)
df['title_t1'] = df['title'].apply(lambda x: 1 if x == 't1' else 0)
print(df) ```
以下に結果を示します:
```title title_t1
0 t1 1
1 t2 0
2 t1 1
3 t1 1
4 t2 0 ```
locを使って処理する
```import pandas as pd
data = {'title': ['t1','t2','t1','t1','t2']}
df = pd.DataFrame(data)
df.loc[df['title'] == 't1', 'title_t1'] = 1
df.loc[df['title'] != 't1', 'title_t1'] = 0
print (df) ```
以下に結果を示します:
```title title_t1
0 t1 1.0
1 t2 0.0
2 t1 1.0
3 t1 1.0
4 t2 0.0 ```