Insert in a list random of 0 and 1 with a certain number of 1's

前端 未结 4 498
别跟我提以往
别跟我提以往 2021-01-16 16:43

How do I generate a list of length N with random 0 and 1 values , but with a given number of 1\'s put randomly in the list.

For example, I want a list of 10

4条回答
  •  隐瞒了意图╮
    2021-01-16 17:05

    Using numpy

    import numpy as np
    from random import shuffle
    A = np.ones(7, dtype=np.int)          // [1 1 1 1 1 1 1] and data-type as integer
    B = np.zeros(3, dtype=np.int)         // [0 0 0] and data-type as integer
    C = np.concatenate((A, B), axis = 0)  // [1 1 1 1 1 1 1 0 0 0]
    shuffle(C)                            // [1 0 0 1 1 1 1 0 1 1]
    print C
    

提交回复
热议问题