如何在乘法计算中处理未定义变量并自动赋值为1

问题描述 #

我有一些预定义变量和现成的乘法公式，一旦存在未定义变量，公式计算就会报错。由于所有公式仅涉及乘法运算，我计划为所有未定义变量赋值为1，如此一来计算就不会中断。请问是否有办法遍历这些公式，识别出未定义变量并将其赋值为1？

以下是示例场景：变量k未定义，导致kbd=k*b*d计算报错：

# 预定义变量
a=10
b=5
c=3
d=6
e=7
f=2
g=9

# 现成公式
abc=a*b*c
fed=d*e*f
efg=e*f*g
kbd=k*b*d

报错信息：

NameError                                 Traceback (most recent call last)
<ipython-input-10-f9e3059809fc> in <module>
----> 1 kbd=k*b*d

NameError: name 'k' is not defined

解决方案 #

这里提供几种实用的实现方式：

方法1：捕获NameError动态赋值 #

直接执行公式，捕获NameError后为未定义变量赋值1，再重新执行公式：

# 预定义变量
a=10
b=5
c=3
d=6
e=7
f=2
g=9

# 公式列表（以字符串形式存储）
formulas = [
    'abc=a*b*c',
    'fed=d*e*f',
    'efg=e*f*g',
    'kbd=k*b*d'
]

# 绑定全局变量环境
global_vars = globals()

for formula in formulas:
    while True:
        try:
            exec(formula, global_vars)
            break
        except NameError as e:
            # 提取报错中的变量名
            var_name = str(e).split("'")[1]
            global_vars[var_name] = 1

# 验证结果
print(kbd)  # 输出：30（1*5*6）

方法2：语法分析提前识别变量 #

通过AST语法分析提前提取公式中的所有变量，为未定义的变量预赋值1：

import ast

# 预定义变量字典
var_env = {
    'a': 10,
    'b': 5,
    'c': 3,
    'd': 6,
    'e': 7,
    'f': 2,
    'g': 9
}

# 公式列表
formulas = [
    'abc=a*b*c',
    'fed=d*e*f',
    'efg=e*f*g',
    'kbd=k*b*d'
]

for formula in formulas:
    # 解析公式的语法树
    tree = ast.parse(formula, mode='exec')
    assign_node = tree.body[0]
    # 提取右侧表达式的所有变量名
    var_names = []
    for node in ast.walk(assign_node.value):
        if isinstance(node, ast.Name):
            var_names.append(node.id)
    # 为未定义变量赋值1
    for var in var_names:
        if var not in var_env:
            var_env[var] = 1
    # 执行公式
    exec(formula, var_env)

# 验证结果
print(var_env['kbd'])  # 输出：30

方法3：字典隔离变量环境（更安全） #

避免修改全局变量，用单独的字典管理公式的变量环境：

var_env = {
    'a': 10,
    'b': 5,
    'c': 3,
    'd': 6,
    'e': 7,
    'f': 2,
    'g': 9
}

formulas = [
    'abc=a*b*c',
    'fed=d*e*f',
    'efg=e*f*g',
    'kbd=k*b*d'
]

for formula in formulas:
    while True:
        try:
            exec(formula, var_env)
            break
        except NameError as e:
            var = str(e).split("'")[1]
            var_env[var] = 1

print(var_env['kbd'])  # 输出：30

注意事项 #

• 以上方法仅适用于纯乘法运算的公式，因为赋值1不会改变乘法的结果逻辑；如果涉及加法、除法等其他运算，赋值1可能导致结果错误。
• 使用exec时需注意代码安全性，避免执行不可信的公式字符串。

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