Python - 列表上的笛卡尔积，其中一些列表中有随机变量

Question

我想在一些列表之间进行笛卡尔积。

某些列表可能包含随机变量，在产品执行时应将这些变量随机化。

我尝试过以下方法。

import random


a_list = ['low', 'medium', 'high']
b_list = [0, 1]

dict_list = []
for a in a_list:
    for b in b_list:
        if a == 'low':
            a_random = random.uniform(0, 3)
        elif a == 'medium':
            a_random = random.uniform(3, 6)
        elif a == 'high':
            a_random = random.uniform(6, 100)
        dict_list.append({'a': a_random, 'b': b})

print(dict_list)

输出：

[{'a': 2.5067206438005165, 'b': 0},
{'a': 2.846737783049243, 'b': 1},
{'a': 4.515841550661135, 'b': 0},
{'a': 5.570169974274982, 'b': 1},
{'a': 26.509898440764896, 'b': 0},
{'a': 57.48321086944802, 'b': 1}]

上面的例子只是一个玩具示例。

我的实际情况会有更多的列表和更多的随机变量。 例如：

# a_list = [random.uniform(0, 3), random.uniform(3, 6), random.uniform(6, 100)]
a_list = ['low', 'medium', 'high'] 
b_list = [0, 1]
# c_list = [random.uniform(0, 10), random.uniform(10, 100), random.uniform(100, 1000)]
c_list = ['low', 'medium', 'high']
d_list = ['A', 'B', 'C', 'D']

是否有更好的方法来达到相同的结果？

也许使用itertool？或发电机？

谢谢！

Answer 1

创建列表后处理随机数。

def lookup1(val):
    if val == 'low':
        return random.uniform(0,3)
    elif val == 'medium':
        return random.uniform(3,6)
    else:
        return random.uniform(6,100)

want = []
for a,b in itertools.product(a_list, b_list):
    want.append( [lookup1(a), b] )

您可能可以对其进行更多参数化。也许创建一个 Distribution 类来保存分布并生成正确的范围。

后续

以下是将分发点嵌入到类中的方法：

class Distribution:
    def __init__( self, limits ):
        self.limits = limits

    def lookup(self, val):
        l = self.limits
        if val == 'low':
            return random.uniform(l[0],l[1])
        elif val == 'medium':
            return random.uniform(l[1],l[2])
        else:
            return random.uniform(l[2],l[3])

distr_a = Distribution( [0, 3, 6, 100] )
distr_b = Distribution( [0, 10, 100, 1000] )

want = []
for a,b in itertools.product(a_list, b_list):
    want.append( [distr_a.lookup(a), b] )

将其扩展为除 ['low','medium','high'] 之外的列表，作为练习留给读者。即使这样也不难。