python - 从满足条件的集合中返回任意元素

Question

有没有办法从满足特定条件的集合中检索任意元素？

例如，如果我的集合有{1,2,3,4,5,6,7,8,9,10}.

有没有办法检索任何小于 5 的任意元素？

标签： pythonpython-3.xset

Answer 1

您可以传递random.choice一个列表，将您的集合过滤为您想要的选择：

from random import choice

s = set([1,2,3,4,5,6,7,8,9,10])
choice([n for n in s if n < 5])

当然，如果你想要所有小于 5 的项目，它只是列表：

[n for n in s if n < 5]