魔杖 sample()函数 - Python

在Python中，我们经常需要随机生成一些数据，例如生成随机数，抽取一定数量的样本等等。为了实现这个功能，Python内置了一个名为 random 的库，提供了丰富的随机生成函数。

其中，sample() 函数可以用于从列表或集合中随机抽取一定数量的元素。它的声明如下：

sample(population, k)

其中，population 是一个列表或集合，表示抽样的总体，k 是一个整数，表示需要抽取的样本数量。

示例：

import random

# 从列表中随机抽取3个元素
lst = ['A', 'B', 'C', 'D', 'E']
sample_lst = random.sample(lst, 3)
print(sample_lst)
# 输出：['C', 'B', 'E']

# 从集合中随机抽取2个元素
set1 = set(range(10))
sample_set = random.sample(set1, 2)
print(sample_set)
# 输出：{9, 6}

如果总体中的元素数量小于需要抽取的样本数量，则会抛出 ValueError 异常。

示例：

# 从长度为3的列表中尝试抽取4个元素，会抛出异常
lst2 = ['A', 'B', 'C']
sample_lst2 = random.sample(lst2, 4)
# 抛出异常：ValueError: Sample larger than population or is negative

综上所述，sample() 函数是一个方便实用的随机抽样函数，可以帮助我们快速地从列表或集合中抽取一定数量的元素。