按照正态分布使用python从列表中选择一个元素
问题描述:
我想按照正态分布使用python从列表中选择一个元素.我有一个列表,例如
I would like to select one element from a list using python following the normal distribution. I have a list, e.g.,
alist = ['an', 'am', 'apple', 'cool', 'why']
例如,根据正态分布的概率密度函数(PDF),给定列表中的第3个元素应该被选择为最大概率.
For example, according to the probability density function (PDF) of normal distribution, the 3rd element in the given list should have the largest probability to be chosen.Any suggestions?
答
from random import normalvariate
def normal_choice(lst, mean=None, stddev=None):
if mean is None:
# if mean is not specified, use center of list
mean = (len(lst) - 1) / 2
if stddev is None:
# if stddev is not specified, let list be -3 .. +3 standard deviations
stddev = len(lst) / 6
while True:
index = int(normalvariate(mean, stddev) + 0.5)
if 0 <= index < len(lst):
return lst[index]
然后
alist = ['an', 'am', 'apple', 'cool', 'why']
for _ in range(20):
print(normal_choice(alist))
给予
why
an
cool
cool
cool
apple
cool
apple
am
am
apple
apple
apple
why
cool
cool
cool
am
am
apple