如何通过从列表中随机选择元素来替换数据框中的值？

Question

我有一个带有 1 和 0 的数据框，它看起来像：

Index Variable_1 Variable_2 Variable_3
A     1          0          1
B     0          1          1
C     0          0          1

我还有一个列表，看起来像：{'X','Y','Z'}

我想用列表中的值随机替换数据框中的所有“1”。 所以输出可能如下所示：

Index Variable_1 Variable_2 Variable_3
A     Y          0          X
B     0          Z          Y
C     0          0          Z

我怎样才能做到这一点？

Answer 1

你可以使用 np.where 和 random.choice 作为

import random

yourlist = ['X','Y','Z']
df = df.where(df == 1, random.choice(yourlist))

您也可以使用 .replace 作为

df = df.replace(1, random.choice(yourlist))

更新：你是对的，我们对每个替换都有相同的 random.choice，它没有对所有元素进行评估？

你能试试这个看看它是如何工作的吗

 df = df.update(np.random.choice(yourlist, size=df.shape), filter_func=lambda x: x==1)