Pandasで複数の選択肢からランダムで文字列を列に入れる

Question

Pandasで複数の選択肢からランダムで文字列を列に入れる

前提

名前
山田
田中
鈴木
佐藤
斎藤

上記に「学年」という列を作成して1年、2年、3年、4年という文字列をランダムに入力したいです。
＊実データではかなりの行数があるので手動でなくコードで解決したいと思っています。

作りたい形

名前	学年
山田	2年
田中	3年
鈴木	3年
佐藤	1年
斎藤	4年

使用データ

入力データ
df = pd.DataFrame(
    data={'名前': ['山田', '田中', '鈴木', '佐藤', '斎藤'], )

試したこと

df['学年']=fakegen.word(ext_word_list=['1年', '2年', '3年', '4年'])

発生している問題

上記のコードでやると下記のように同じ値のみ出力されてしまいます。

名前	学年
山田	2年
田中	2年
鈴木	2年
佐藤	2年
斎藤	2年

Answer 1

import pandas as pd
import random

df = pd.DataFrame({
    '名前': ['山田', '田中', '鈴木', '佐藤', '斎藤']
})

grade = ['1年', '2年', '3年', '4年']
df['学年'] = random.choices(grade, k=len(df), weights=[1]*len(grade))
print(df)

Answer 2

random.choiceで必要な数だけ乱数で選択してやればよいです。

import pandas as pd
import random

df = pd.DataFrame(['山田','田中','鈴木','佐藤','斎藤'], columns=['名前'])

l = [f'{i+1}年生' for i in range(4)]
df['学年'] = pd.Series([random.choice(l) for _ in range(len(df))])
print(df)
"""
   名前   学年
0  山田  2年生
1  田中  1年生
2  鈴木  4年生
3  佐藤  2年生
4  斎藤  4年生
"""