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

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

1 个答案: