双变量数学表达式输入、求值及结果存储功能需求

功能说明 #

核心执行流程：

• 引导用户输入变量X、Y的数值
• 接收用户输入的含X、Y的表达式（支持2X、Y+3X^2、√(X+Y)这类简化写法）
• 解析并计算表达式结果
• 存储运算结果

代码实现（Python） #

import math
import re

def preprocess_expression(expr):
    # 处理数字紧跟变量的情况：2X → 2*X，3Y → 3*Y
    expr = re.sub(r'(\d+)([XY])', r'\1*\2', expr)
    # 处理变量紧跟数字的情况（兼容非常规写法）：X2 → X*2
    expr = re.sub(r'([XY])(\d+)', r'\1*\2', expr)
    # 替换幂运算符号^为Python标准的**
    expr = expr.replace('^', '**')
    # 替换根号√为math.sqrt，适配括号包裹的运算内容
    expr = re.sub(r'√\((.*?)\)', r'math.sqrt(\1)', expr)
    return expr

# 获取用户输入
x = float(input("请输入变量X的取值："))
y = float(input("请输入变量Y的取值："))
expression = input("请输入以X、Y为变量的表达式（示例：2X、Y+3X^2、√(X+Y)）：")

# 预处理表达式为可执行语法
processed_expr = preprocess_expression(expression)

# 计算并存储结果
try:
    # 限制eval的可用环境，仅开放安全的函数和变量
    result = eval(processed_expr, {"__builtins__": None, "math": math, "X": x, "Y": y})
    print(f"运算结果O：{result}")
    
    # 示例存储方式：追加写入文本文件，可按需改为数据库/内存存储
    with open("expression_result.txt", "a", encoding="utf-8") as f:
        f.write(f"X={x}, Y={y}, 表达式={expression}, 结果={result}\n")
    print("结果已成功存储到expression_result.txt")
except Exception as e:
    print(f"表达式计算出错：{str(e)}")

关键细节说明 #

• 表达式预处理：通过正则表达式转换用户输入的简化写法为Python可识别的语法，解决2X、^、√这类符号的兼容问题
• 安全求值：使用eval时严格限制可用的内置函数和模块，仅开放math模块和自定义的X/Y变量，避免恶意表达式风险
• 结果存储：示例采用追加写入文本文件的方式，可根据实际需求修改为存入数据库、内存变量等形式

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