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

前提

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

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

作りたい形

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

使用データ

python

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

試したこと

python

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

発生している問題

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

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

コメントを投稿

0 コメント