У меня простая проблема. Мне нужен способ сделать функцию, которая генерирует 0 в p процентах случаев и 1 во всех остальных случаях. Я попытался сделать это с random.random (), как это:

p = 0.40

def generate():
    x = random.random()
    if x < p:
        return 0
    else:
        return 1

Тем не менее, это некажется хорошим способом. Или это так?