タイタニックのデータに、カラムを追加したいです。

Question

添付のようなTitanicのデータフレームの「age」のデータから判定し、子供、大人、シニアに分類した新たなカラム「generation」をデータフレームに追加したいです。

関数を作成し、それをカラム「age」に適用させて、新たなカラム「generation」に代入する形で考えてみたのですが、
データフレームの1行1行判定していくので、for文を入れた方が良いのでしょうか…。
その辺りが苦手なので、よくわかりません。

どなたか教えていただけますと幸いです。

def age_check(data):
    if age < 20:
        return "child"
    elif 20 <= age < 60:
        return "adult"
    else:
        return "senior"

train['generation'] = age_check(train['age'])

Answer 1

Pandasのapplyを使います。

train['generation'] = train['age'].apply(age_check)

ちなみに、def age_check(data):の引数dataはageにする必要があると思います。

Comments

[sakanamono]

無事カラムを作成できました！ありがとうございました！