向列表列表中的项目添加概率

Question

我想为列表中的每个项目添加概率，而该列表在另一个列表中。

一些伪代码：

myList = [ [a, b, c, d], [e, f, g, h], [i, j, k, l], [m, n, o], [p, q, r], [s, t, u] ]

probabilities = [ [0.6, 0.3, 0.075, 0.025], [0.6, 0.3, 0.075, 0.025], [0.6, 0.3, 0.075, 0.025], [0.55, 0.35, 0.1], [0.55, 0.35, 0.1], [0.55, 0.35, 0.1] ]

有没有办法做到这一点？

进一步：

我的需要是创建另一个类似于下面的列表...

newList = [ [b, e, k, o, p, s], [a, f, i, m, r, t], ... etc. ] 

在给定概率的情况下，随机选择每个元素，并且 newList 中没有两个列表是相同的。我不确定这是否可以实现。

到目前为止我的代码：

layers = [list(Path(directory).glob("*.png")) for directory in ("dir1/", "dir2/", "dir3/", "dir4/", "dir5/", "dir6/")]

list_of_prob = [[0.6, 0.3, 0.075, 0.025], [0.6, 0.3, 0.075, 0.025], [0.6, 0.3, 0.075, 0.025], [0.6, 0.3, 0.1], [0.6, 0.3, 0.1], [0.6, 0.3, 0.1]]

rwp = [choices(layers, list_of_prob, k=????)]

rand_combinations = [([choice(k) for k in layers]) for i in choice_indices]

我不完全确定choices() 中的k 是什么，例如。列表的数量或列表中的总元素数。 Layers 是图片路径的列表，.pngs，与上面伪代码中提供的“myList”格式相同（dir1 中 4 个图像，dir2 中 4 个图像，dir3 中 4 个图像，dir4 中 3 个，dir5 中 3 个， 3 在 dir6)。

我已经有代码可以遍历列表并创建随机图像，但我希望仅在 x% 的时间内生成一些图像。因此，我原来的问题。对不起，如果我只是把事情复杂化了，我试着简化它。

Answer 1

我将myList 转换为字符串只是为了方便。

这将创建组合并将它们附加到newList，而忽略newList中已经存在的任何组合

当newList的长度等于myList的长度时循环结束

import random

myList = [['a', 'b', 'c', 'd'], 
          ['e', 'f', 'g', 'h'], 
          ['i', 'j', 'k', 'l'], 
          ['m', 'n', 'o'], 
          ['p', 'q', 'r'], 
          ['s', 't', 'u']]

probabilities = [[0.6, 0.3, 0.075, 0.025], 
                 [0.6, 0.3, 0.075, 0.025], 
                 [0.6, 0.3, 0.075, 0.025], 
                 [0.55, 0.35, 0.1], 
                 [0.55, 0.35, 0.1], 
                 [0.55, 0.35, 0.1]]
newList = []

def random_list():
    combo = []
    for d, p in zip(myList, probabilities):
        choice = random.choices(d, p)
        combo.append(''.join(choice))
    return combo

while len(newList) < len(myList):
    a = random_list()
    if a not in newList:
        newList.append(a)

newList 的结果：

[['b', 'f', 'k', 'm', 'q', 's'],
 ['a', 'e', 'j', 'm', 'q', 't'],
 ['b', 'f', 'k', 'm', 'q', 'u'],
 ['a', 'f', 'i', 'm', 'p', 's'],
 ['a', 'e', 'i', 'n', 'p', 't'],
 ['b', 'f', 'k', 'm', 'r', 'u']]

Answer 2

因此，如果我理解正确，您希望根据概率从myList 的每个子列表中获取一个元素以获取一个元素，并多次执行此操作以获取列表列表。

使用 Python 3，您可以：

import random
# the new set since two elements cannot be the same
result_set = set()
# the number of times you want to do it, you can change it.
for i in range(5):
    # A single result as a str, hashable.
    result = ""
    for chars, char_probabilities in zip(myList, probabilities):
    # weighted random choice
    char = random.choices(chars, char_probabilities)
    result += char[0]
    result_set.add(result)

result_list = [list(elt) for elt in result_set]