问题描述
从列表中,我必须将 quantity
大小的随机元素列表转换为另一个名为 sample
的列表。这就是我所拥有的:
used = [element1.png,element2.png]
sample = random.sample(os.listdir(current_dir + '/logos/'),quantity)
我如何确保这个随机样本没有来自 used
的任何项目?是否可以在一行中执行此操作,而不是庞大的 if 语句?
解决方法
把你的人口变成一个集合,然后用 set(used) 取差。然后从差异结果中抽取样本。
sample = random.sample(
list(set(os.listdir(current_dir + '/Logos/')) - set(used)),quantity)
,
IIUC 或许可以尝试一个简单的 list
理解:
used = ['element1.png','element2.png']
sample = random.sample([i for i in os.listdir(current_dir + '/Logos/') if i not in used],quantity)
现在:
print(sample)
会给出预期的结果。