python - 在具有随机字符串值的 Pandas 中插入一个新列

Question

我有一个数据框

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

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

     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，但有更简单的方法。

Answer 1

您必须将新列设置为 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 分别设置选择"is"或“否”的概率。如果您希望获得"is"的机会更高，即您必须增加第一个数字。

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

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