代码之家  ›  专栏  ›  技术社区  ›  Carrot Man

如何让熊猫数据框随机填充1和0?

  •  0
  • Carrot Man  · 技术社区  · 2 年前

    我有一个 pandas dataframe 在python中,我想根据随机选择填充1和0。 如果可能的话,我希望能和numpy一起完成。我试过了

    tw['tw'] = np.random.choice([0, 1])
    

    但这最终给了我一个机会 dataframe 填充了1或0。我知道使用for循环是可能的:

    for i in range(len(tw)):
        tw['tw'][i] = np.random.choice([0, 1])
    

    但这感觉效率很低。我该怎么做呢?

    2 回复  |  直到 2 年前
        1
  •  2
  •   mozway    2 年前

    如果指定标量,该值将广播到所有索引。

    你需要分配 序列大小的数组 .

    使用 size 的参数 numpy.random.choice :

    tw['tw'] = np.random.choice([0, 1], size=len(tw))
    
        2
  •  2
  •   Vishal Balaji    2 年前

    你可以用numpy的 random integer generator

    tw['tw'] = np.random.randint(low = 0,high = 2,size = len(tw))
    

    请注意,“高”数字不包括在内,因此必须给出2。