在Python中从列表字典中选择随机项

Question

我试图从一个列表字典中随机选择一个特定数量的序列，然后将这些序列附加到一个新的列表中，但是我一直收到一个空列表。我的代码：

final_List = []
num = [["ENST", "10", "5"], ["ENGT", "8", "2"], ["ENHT", "5", "1"]]
a = {"ENST" : ['acac', 'tgtgtgt', 'hahah'], "ENHT": ['aaaa', 'tttt', 'gig', 
'cccc'], 'ENGT' : ['ddd', 'eeee', 'ffff', 'hhhh', 'pppp']}

for line in num:
    for k, v in a.items():
        if line[0] == k:
            for i in int(line[1] -1):
                final_List.append([k, random.choice(0, len(v))])

我将[0]行与k行进行比较，如果它们相同，我想从v中提取随机序列。我想要的序列数是[1]行的数字。最终的清单如下所示：

final_List = [["ENST", "acac"], ["ENST", tgtgtgt"]...10 lists with header "ENST"
              ["ENGT", "ddd"], .....8 lists with header "ENGT"
              ["ENHT","aaaa"], ..... 5 lists with header "ENHT"

Answer 1

类似于列表理解的想法，但我会使用一个函数来让它更清晰一点

def create_sequence(hdr, count):
     return [[hdr, random.choice(a[hdr])] for _ in range(count)]

final_list = [create_sequence(hdr, int(count)) for hdr,count,_ in num]

Answer 2

问题出在你随机选择的那条线上。您使用random.choice不正确。另外，我们要确保您在第[1]行上进行的迭代是正确的。线路：

for i in int(line[1] -1):
    final_List.append([k, random.choice(0, len(v))])

应该是

for i in range(int(line[1])):
    final_List.append([k, random.choice(v)])

Answer 3

您可以使用嵌套列表理解：

>>> import pprint
>>> import random
>>> num = [["ENST", "10", "5"], ["ENGT", "8", "2"], ["ENHT", "5", "1"]]
>>> a = {"ENST" : ['acac', 'tgtgtgt', 'hahah'], "ENHT": ['aaaa', 'tttt', 'gig', 
... 'cccc'], 'ENGT' : ['ddd', 'eeee', 'ffff', 'hhhh', 'pppp']}
>>> res = [[[k, random.choice(a[k])] for _ in range(int(count))] for k, count, _ in num]
>>> pprint.pprint(res)
[[['ENST', 'acac'],
  ['ENST', 'tgtgtgt'],
  ['ENST', 'hahah'],
  ['ENST', 'hahah'],
  ['ENST', 'tgtgtgt'],
  ['ENST', 'tgtgtgt'],
  ['ENST', 'hahah'],
  ['ENST', 'acac'],
  ['ENST', 'acac'],
  ['ENST', 'tgtgtgt']],
 [['ENGT', 'pppp'],
  ['ENGT', 'ffff'],
  ['ENGT', 'eeee'],
  ['ENGT', 'hhhh'],
  ['ENGT', 'pppp'],
  ['ENGT', 'eeee'],
  ['ENGT', 'pppp'],
  ['ENGT', 'pppp']],
 [['ENHT', 'cccc'],
  ['ENHT', 'aaaa'],
  ['ENHT', 'gig'],
  ['ENHT', 'tttt'],
  ['ENHT', 'cccc']]]

在上面for k, count, _ in num将迭代num中的元素，并将它们解压到相应的变量中。请注意，我们只需要名称和计数，因此一次性变量的名称为_：

>>> [(k, count) for k, count, _ in num]
[('ENST', '10'), ('ENGT', '8'), ('ENHT', '5')]

对于num中的每个元素，通过使用^{}从a中的相应列表中选择count元素来构造一个列表：

>>> k = 'ENST'
>>> count = '10'
>>> [[k, random.choice(a[k])] for _ in range(int(count))]
[['ENST', 'tgtgtgt'], ['ENST', 'hahah'], ['ENST', 'acac'], ['ENST', 'hahah'], ['ENST', 'tgtgtgt'], ['ENST', 'hahah'], ['ENST', 'hahah'], ['ENST', 'hahah'], ['ENST', 'acac'], ['ENST', 'acac']]