titanic 新しいカテゴリーを作り、数値を代入したい

trainのAgeを新しい変数を作り、数値を代入したい。下記の3つのグラフから20～３０代の年齢の人がPclass=3の人の生存率が高く、特徴量？を作りたいと思った。train['new_age'].unique()をすると0,1,2,3とでなかった。
1下記のコードが上手くいってない理由
2下記のコードでif文で書くならばどうなりますか？（を作り、数値を代入したい。下記の3つのグラフから20～３０代の年齢の人がPclass=3の人の生存率が高く、特徴量？を作りたいと思った。train['new_age'].unique()をすると0,1,2,3とでなかった。
1下記のコードが上手くいってない理由
2下記のコードでif文で書くならばどうなりますか？（ValueError: The truth value of a Series is ambiguous. Use a.empty, a.bool(), a.item(), a.any() or a.all().）になりできなかった。
3特徴量を作るときにアドバイスがあったらお願いします。（任意）

コード

import pandas as pd 
import matplotlib.pyplot as plt
import seaborn as sns
train = pd.read_csv('train.csv')

fig=plt.figure(figsize=(18,6))
sns.histplot(x=train['Age'], shrink=0.8, bins=12, hue=train['Survived'], multiple="dodge", palette='dark')
plt.show()
fig=plt.figure(figsize=(10,6))
sns.countplot(x='Survived',hue=train['Pclass'],data=train)
fig=plt.figure(figsize=(18,6))
sns.histplot(x=train['Age'], shrink=0.8, bins=12, hue=train['Pclass'], multiple="dodge", palette='dark')
plt.show()

train['new_age'] = 0 
train.loc[(train['Age']<20),'new_age'] = 1
train.loc[(train['Age']<40),'new_age'] = 2
train.loc[(train['Age']<50),'new_age'] = 3
train['new_age'].unique()

行動規範の内容に同意します

回答1件

ベストアンサー

1 下記のコードが上手くいってない理由

レベル化の順序が逆だから。

python
1import pandas as pd 
2
3titanic = pd.read_csv('titanic.csv')
4
5titanic['new_age'] = 0 
6titanic.loc[(titanic['Age']<50),'new_age'] = 3
7titanic.loc[(titanic['Age']<40),'new_age'] = 2
8titanic.loc[(titanic['Age']<20),'new_age'] = 1
9
10print(titanic['new_age'].unique())
11
12#
13[2 0 1 3]

以下、pandas.cut を使う方法。

python
1titanic['new_age'] = (
2  pd.cut(titanic['Age'].fillna(99), bins=[0, 20, 40, 50, 100],
3         right=False, labels=False)
4    .map({0:1, 1:2, 2:3, 3:0}))