按照正态分布使用python从列表中选择一个元素

Fra*_*ang 1 python pdf select distribution list

我想按照正态分布使用python从列表中选择一个元素。我有一个清单,例如

alist = ['an', 'am', 'apple', 'cool', 'why']
Run Code Online (Sandbox Code Playgroud)

例如,根据正态分布的概率密度函数(PDF),给定列表中的第3个元素应该被选择为最大概率。

Hug*_*ell 5

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]
Run Code Online (Sandbox Code Playgroud)

然后

alist = ['an', 'am', 'apple', 'cool', 'why']
for _ in range(20):
    print(normal_choice(alist))
Run Code Online (Sandbox Code Playgroud)

why
an
cool
cool
cool
apple
cool
apple
am
am
apple
apple
apple
why
cool
cool
cool
am
am
apple
Run Code Online (Sandbox Code Playgroud)