如何用Python或SQLite实现多表查询满足A-B+C<200的组合？

给你两种可行的实现方式，覆盖不同数据量场景：

方法一：Pandas 交叉连接+筛选 #

先把三个数组转成DataFrame，生成所有可能的三元组组合，再筛选符合条件的结果，逻辑直观易懂：

import pandas as pd

# 构造数据
df_a = pd.DataFrame({'A': [1, 8]})
df_b = pd.DataFrame({'B': [10, 23, 15, 80]})
df_c = pd.DataFrame({'C': [500, 200, 180]})

# 生成所有三元组的笛卡尔积
all_combinations = df_a.merge(df_b, how='cross').merge(df_c, how='cross')

# 筛选满足 A-B+C < 200 的组合
result = all_combinations[all_combinations['A'] - all_combinations['B'] + all_combinations['C'] < 200]

print(result)

运行后会直接输出所有符合条件的(A,B,C)三元组，适合数据量不大的场景。

方法二：NumPy 广播机制 #

利用NumPy的广播特性，不用生成全量组合就能快速定位符合条件的三元组，效率更高：

import numpy as np

# 构造数组
a = np.array([1, 8])
b = np.array([10, 23, 15, 80])
c = np.array([500, 200, 180])

# 通过广播扩展维度，实现逐元素计算
a_3d = a[:, np.newaxis, np.newaxis]
b_3d = b[np.newaxis, :, np.newaxis]
c_3d = c[np.newaxis, np.newaxis, :]

# 计算满足条件的位置
condition = (a_3d - b_3d + c_3d) < 200

# 获取符合条件的索引并提取对应数值
indices = np.where(condition)
result = np.column_stack((a[indices[0]], b[indices[1]], c[indices[2]]))

print(result)

这种方法避免了生成大量中间数据，适合处理规模较大的数据集。

内容的提问来源于stack exchange，提问作者mike