Question

我有一份清单词典：

lists=dict(animals=["dog","cat","shark"],
           things=["desk","chair","pencil"],
           food=["spaghetti","ice-cream","potatoes"])

如何让Python从其中一个列表中随机选择一个项目并告诉我它所在的列表？或者我如何从字典中选择一个键，然后从与该键对应的列表中选择一个值？

例如：

 
dog - from animals
potatoes - from food

Answer 1

random.choice从序列中选择一个随机项：

import random

从您dict lists中选择要绘制的密钥：

which_list = random.choice(lists.keys())

然后，使用该密钥从list获取dict：

item = random.choice(lists[which_list])

如果你需要相同的权重：

import random

which_list, item = random.choice([(name, value) 
                                     for name, values in lists.iteritems() 
                                         for value in values])

Answer 2

我可以立即想到的两种方法：

首先选择一个列表名称（键），然后从中选择一个条目 - 如果列表有不同的长度，你只需要小心，如果你想要一个统一的分布
将列表的dict压缩成一个('list-name','value')对的列表（无论每个列表有多少条目，但需要更多内存，更容易获得统一分布权）

做前者的一种方法：

from itertools import chain
import random
weight_choices = list(chain(*([name] * len(values) for (name, values) in lists.iteritems()))) # generate a list of the form ("animals", "animals", "animals", ...)
list_name = random.choice(weight_choice) # The list it's chosen from...
chosen_item = random.choice(lists[list_name]) # and the item itself

（如果你不关心在列表之间进行统一分配：）

import random
list_name = random.choice(lists.keys())
chosen_item = random.choice(lists[list_name])

......和后一种方法：

from itertools import chain, repeat
all_items = list(chain(*((zip(repeat(name), values) for (name, values) in lists.iteritems()))))
list_name, chosen_item = random.choice(all_items)

以及后者的itertools方式：

all_items = []
for name, values in lists.iteritems():
  for value in values:
    all_items.append((name, value))
list_name, chosen_item = random.choice(all_items)

如何从字典中的某个列表中随机选择一个项目，并获取列表的密钥？

2 个答案: