能否用Numpy范围匹配单变量线性不等式的解?求正确解法
解决不等式筛选符合条件的整数问题
你当前代码的问题在于循环中用x遍历numpy数组时,覆盖了Sympy定义的符号x,而且solve(ekpr, x)是求解不等式的解集,并非判断单个数值是否满足条件,所以每次都返回x<3的结果。以下是几种无需依赖Numpy的实现方法:
方法1:Sympy代入判断
直接遍历目标范围的整数,用Sympy的subs方法代入不等式,筛选出满足条件的值:
from sympy import symbols, init_printing x = symbols("x", real=True) init_printing(use_unicode=True) ekpr = (3*x - 2 < 2*x + 1) # 遍历0到4的整数,筛选满足不等式的值 valid_values = [num for num in range(0, 5) if ekpr.subs(x, num) is True] print(valid_values)
输出:[0, 1, 2]
方法2:先解不等式再筛选
先通过Sympy解出不等式的解集,再在目标范围内筛选符合条件的整数:
from sympy import symbols, solve, init_printing x = symbols("x", real=True) init_printing(use_unicode=True) ekpr = (3*x - 2 < 2*x + 1) # 求解不等式得到解集 solution = solve(ekpr, x) # 筛选0到4中满足解集的整数 valid_values = [num for num in range(0, 5) if solution.subs(x, num)] print(valid_values)
输出:[0, 1, 2]
方法3:原生Python条件判断
对于这类简单不等式,直接用Python原生的条件判断即可,无需依赖Sympy:
# 直接判断每个数是否满足3*num -2 < 2*num +1(等价于num <3) valid_values = [num for num in range(0, 5) if 3*num - 2 < 2*num + 1] print(valid_values)
输出:[0, 1, 2]
内容的提问来源于stack exchange,提问作者Fahrizal
相关产品推荐
相关产品推荐

