关于随机数和固定的问题

Question

请问如何生成有n个数的数组，里边只有0，1，-1排列随机，这个我知道用random.randint(-1,2,n)，并且数组所有数加起来和为0呢？这一步怎么写代码呢？

Answer 1

该回复引自免费微信小程序-皆我百晓生:

---

生成数组的代码可以写成：

import random

n = 10 # 假设生成长度为10的数组

arr = [random.choice([-1, 0, 1]) for _ in range(n)]

这样我们就可以生成一个长度为n的数组arr，其中的数只有0，1，-1。

要确保数组中所有数加起来的和为0，我们可以在生成数组后进行调整，将其中一个数替换为相反数。具体代码如下：

import random

n = 10 # 假设生成长度为10的数组

arr = [random.choice([-1, 0, 1]) for _ in range(n)]

# 计算数组总和
total_sum = sum(arr)

# 判断总和是否为0
if total_sum != 0:
    # 随机选择一个非零元素，并替换为相反数
    non_zero_indices = [i for i in range(n) if arr[i] != 0]
    random_index = random.choice(non_zero_indices)
    arr[random_index] *= -1

这样就确保了数组中所有数加起来的和为0。