python - 创建一列并随机赋值

Question

我有一个包含客户 ID 的数据框。

我想创建一个名为 group_user 的新列，它只需要 3 个值:0,1,2

我希望这些值以平衡的比例随机分配给客户。

输出将是:

ID         group_user
341          1
127          0 
389          2

谢谢!

Answer 1

你可以试试这个:

>>> lst = [0, 1, 2]
>>> df['group_user'] = pd.Series(np.tile(lst, len(df) // len(lst) + 1)[:len(df)]).sample(frac=1)
>>> df

这适用于所有长度的列和列表。