从 Python 中的集合中获取随机元素

要从集合中获取随机元素:

  1. 使用 list() 类将集合转换为列表。
  2. 使用 random.choice() 方法获取随机元素。
  3. random.choice() 方法从序列中返回一个随机元素。
import random

my_set = {'zadmei', 'www', '.', 'com'}

result = random.choice(list(my_set))
print(result)  # 👉️ 'com'

result = random.sample(list(my_set), k=2)
print(result)  # 👉️ ['www', 'zadmei']

我们使用 list() 类将集合对象转换为列表。

这是必要的,因为集合对象是元素的无序集合,不支持随机访问,因此我们不能直接从集合中获取随机元素。

如果我们需要优化性能,请将集合转换为元组。

import random

my_set = {'zadmei', 'www', '.', 'com'}

result = random.choice(tuple(my_set))
print(result)  # 👉️ 'com'

result = random.sample(tuple(my_set), k=2)
print(result)  # 👉️ ['.', 'zadmei']

random.choice 方法接受一个序列并从非空序列中返回一个随机元素。

import random

print(random.choice(['zadmei', 'www'])) # 👉️ "hadz"

如果序列为空,则该方法引发 IndexError

如果需要从集合中获取多个随机元素,可以使用 random.sample 方法。

import random

my_set = {'zadmei', 'www', '.', 'com'}

result = random.sample(tuple(my_set), k=2)
print(result)  # 👉️ ['com', 'zadmei']

从 Python 中的集合中获取随机元素

random.sample 方法返回从提供的序列中选择的 N 个唯一元素的列表。

该方法采用的第一个参数是一个序列,第二个参数是要返回的随机元素的数量。

import random

# 👇️ [2, 4]
print(random.sample([1, 2, 3, 4, 5], k=2))

random.sample() 方法返回从序列中选择的 N 大小的元素列表,无需替换。

没有替换意味着不能多次返回相同的元素。

注意 ,如果我们尝试选择的元素多于序列中存在的元素,则 random.sample() 方法会引发 ValueError

import random

my_set = {'zadmei', 'www', '.', 'com'}

# ⛔️ ValueError: Sample larger than population or is negative
result = random.sample(tuple(my_set), k=5)

从 Python 中的集合中获取随机元素

如果需要从集合中选择多个随机元素,也可以使用 random.choices() 方法。

import random

my_set = {'zadmei', 'www', '.', 'com'}

result = random.choices(tuple(my_set), k=2)

print(result)  # 👉️ ['com', 'www']

random.choices 方法返回一个 k 大小的元素列表,该列表是从提供的可替换的迭代中选择的。

替换基本上意味着可以多次返回相同的元素。

import random

my_set = {'zadmei', 'www', '.', 'com'}

result = random.choices(tuple(my_set), k=5)

print(result)  # 👉️ ['zadmei', '.', 'com', 'www', '.']

从 Python 中的集合中获取随机元素

如果可迭代对象为空,则该方法会引发 IndexError 异常。

random.choices() 和 random.sample() 方法的主要区别在于:

  • random.choices() 选择替换元素
  • random.sample() 选择没有替换的元素