在 pandas 中插入具有随机字符串值的新列

我有一个数据框

     A B C
   0 1 2 3  
   1 2 3 3  
   2 3 2 1  

我需要在 pandas DataFrame 中创建一个新列，并随机填充此列中的“是”或“否”。

     A B C  NEW
   0 1 2 3  yes
   1 2 3 3  no
   2 3 2 1  no

使用 random.choice 会导致每一行的结果都相同的列

     A B C  NEW
   0 1 2 3  no
   1 2 3 3  no
   2 3 2 1  no

我尝试了 map、apply 和 applymap 但有更简单的方法。

您必须将新列设置为pd.Series然后使用random.choices:

import random

df['NEW'] = pd.Series(
    random.choices(['yes', 'no'], weights=[1, 1], k=len(df)), 
    index=df.index
)

random.choices将为每一行选取其中一个值。

weights分别设置选择“是”或“否”的概率。如果您希望获得更高的“是”机会，即您必须增加第一个数字。

k设置系列的长度。它必须具有相同长度的 DataFrame。

index重要的是设置为相同df.index否则，无论您是否从更大的 DataFrame 中切片它，它都可以填充 NaN