本文详解如何在数据库（sql）和编程语言（python）中，根据表中 `weight` 列对行进行加权随机抽样，确保每行被选中的概率严格正比于其权重值。
在数据分析与实验设计中，常需从非均匀分布的候选集中按权重进行随机抽样——例如 A/B 测试中的流量分配、推荐系统中的多样性采样，或蒙特卡洛模拟中的重要性抽样。核心要求是：某行被抽中的概率 = 该行 weight / 所有 weight 之和。以示例数据为例：
| view | weight |
|---|---|
| A | 1 |
| B | 1 |
| C | 2 |
| D | 1 |
| E | 1 |
| F | 1 |
| G | 3 |
总权重为 1+1+2+1+1+1+3 = 10，因此 C 的抽中概率为 2/10 = 0.2，G 为 3/10 = 0.3，其余均为 0.1。
✅ SQL 实现（通用思路：权重展开 + 随机排序）
主流关系型数据库（如 MySQL、PostgreSQL、SQLite）不直接支持 TABLESAMPLE SYSTEM (n) WITH WEIGHTS（目前仅 PostgreSQL 16+ 实验性支持），但可通过「权重展开」技巧稳健实现：
SELECT t.* FROM mytable t INNER JOIN ( SELECT 1 AS weight UNION ALL SELECT 2 UNION ALL SELECT 3 ) d ON d.weight <= t.weight ORDER BY RANDOM() -- PostgreSQL / SQLite -- ORDER BY RAND() -- MySQL LIMIT 1;
⚠️ 注意：上述写法中子查询 (SELECT 1 UNION ALL SELECT 2 UNION ALL SELECT 3) 是手动枚举最大权重值（3）的自然数序列。若最大权重较大（如 1000），手动枚举不可行。此时推荐使用递归 CTE（PostgreSQL/SQL Server）或生成数字序列的辅助表。更健壮的替代方案是使用窗口函数 + 累计权重 + 随机浮点数二分查找（适用于大数据量），但复杂度显著上升。
立即学习“Python免费学习笔记（深入）”；
✅ 优势：纯 SQL、无需应用层逻辑、可嵌入视图或存储过程。
⚠️ 局限：展开后中间结果集可能膨胀（如某行 weight=1000，则生成 1000 行副本），内存与性能需评估。
✅ Python 实现（使用 pandas + numpy）
若数据已加载至内存，推荐使用 pandas.DataFrame.sample() 的 weights 参数，底层调用 numpy.random.Generator.choice，高效且语义清晰：
import pandas as pd
import numpy as np
# 构造示例数据
df = pd.DataFrame({
'view': ['A', 'B', 'C', 'D', 'E', 'F', 'G'],
'weight': [1, 1, 2, 1, 1, 1, 3]
})
# 按 weight 列加权抽样（replace=False 表示无放回；此处通常设 True 或省略）
sampled = df.sample(n=1, weights='weight', random_state=42)
print(sampled)
# view weight
# 6 G 3也可直接使用 NumPy 进行底层控制：
rng = np.random.default_rng(42) idx = rng.choice(df.index, p=df['weight'] / df['weight'].sum()) print(df.iloc[idx])
✅ 优势：简洁、可复现（通过 random_state）、天然支持批量抽样（n=100）、自动归一化权重。
⚠️ 注意：weights 列必须全为非负数，且不能全为零；若含 NaN，需提前填充或过滤。
? 关键总结与最佳实践
- 概率保真性：两种方法均严格满足 $P(\text{row}_i) = \frac{w_i}{\sum w_j}$，前提是权重为非负实数。
- 性能权衡：SQL 展开法适合中小权重值（≤100）及小表；Python 更适合预加载场景或需后续链式处理的流程。
-
扩展建议：
- 若需无放回多行抽样（如抽 3 个互异 view），SQL 中需改用 ROW_NUMBER() OVER (ORDER BY RANDOM()) + 权重展开后去重；Python 中直接设 replace=False 即可。
- 若权重为浮点数或精度敏感（如金融场景），Python 中建议显式归一化并验证 np.isclose(weights.sum(), 1.0)。
- 安全提醒：避免在 SQL 中拼接用户输入构造权重序列，防止注入；Python 中注意 weights 向量长度须与 DataFrame 行数一致。
掌握加权抽样的原理与实现，是构建可靠数据管道与概率算法的基础能力。无论选择 SQL 原生方案还是 Python 生态工具，理解“权重→概率映射”这一本质，方能灵活适配各类业务约束。
