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 ```