將 1 的隨機數插入一列 Pandas 數據框 python

Question

我有一個 182 行的數據框。 我想向其中添加一個充滿 0 的列，除了 df 中隨機分布的 12 個 1。 是否可以？ 謝謝

Answer 1

df['x']=0
df.loc[df.sample(12).index.tolist(), 'x'] =  1

Answer 2

您可以為此使用random.shuffle

# create members
contents = [1 for _ in range(12)] + [0 for _ in range(182 - 12)]
# shuffle contents in-place
random.shuffle(contents)
# add to df
df["colname"] = contents

Answer 3

您可以將random.choice與列表理解一起使用：

import random
df['NEW_COL'] = [0]*182
df['NEW_COL'][[random.choice(range(182)) for x in range(12)]] = 1