生成numpy中的随机二进制数组，该数组的概率由索引给定

import numpy as np

def generate_binary_array(probabilities):
    # 确定数组长度
    array_length = len(probabilities)

    # 生成随机数数组
    random_numbers = np.random.rand(array_length)

    # 根据索引和概率生成二进制数组
    binary_array = np.zeros(array_length, dtype=int)
    for i in range(array_length):
        if random_numbers[i] < probabilities[i]:
            binary_array[i] = 1

    return binary_array

probabilities = [0.2, 0.8, 0.5, 0.3, 0.9]
binary_array = generate_binary_array(probabilities)
print(binary_array)

[0 1 0 0 1]