创建一个总和为特定数字的数组？

Question

我正在使用python.我正在尝试创建一个总和为某个数字的数组,但我希望数组的长度与我选择的一样.有关如何做到这一点的任何想法？例如,我想要一个总和为1的数组.所以[.25,.25,.25,.25],但我希望数组中的数字是随机的,我希望我的数组有一个长度98.有人可以帮忙吗？

Answer 1

这是一个简单的实现,通过构建一个包含98个随机数的数组,找到它们的总和,然后将每个数字除以总和,使整个数组归一化为1.

import random

# generate a random array of 98 numbers
numbers = [random.random() for _ in range(98)]
print(numbers) 
# [0.2378205280188267, 0.08942239291741982, ...]
print(sum(numbers))
# 48.8051742287

# normalise the array to sum to 1
normalised = [r / sum(numbers) for r in numbers]
print(normalised) 
# [0.004872854810521523, 0.0018322318141581963, ...]
print(sum(normalised)) 
# 1.0

但是如果你正在进行大量的计算,你应该考虑使用NumPy.

编辑:上面的等效代码,但更容易阅读,没有列表理解:

# build a list/array of random numbers
numbers = []
for _ in range(98):
    numbers.append(random.random())

# get the sum of the array
print(sum(numbers))  
# 48.8051742287

# normalise the original list using the sum
normalised = []
for n in numbers:
    normalised.append(n / sum(numbers))
print(sum(normalised)) 
# 1.0