python - 使用 np.random 生成值

Question

我尝试向 df 添加新列，其中所有值都为 0 或 1。我用

df['name'] = np.random.randint(0, 1)

但它返回所有 0 的列。

Answer 1

使用形状说明符让它在区间 [0,2) 中创建一个随机整数数组，即 0 或 1长度与 dataframe 中的行数相同，np.random.randint(0, 2,(df.shape[0])) 然后指定为新专栏。因此，实现将是 -

df['name'] = np.random.randint(0, 2,(df.shape[0]))