Discord.py机器人如何提取消息中的数值并处理数学运算？

针对你需要从Discord消息中提取整数/浮点数、支持数学运算并忽略无关文本的需求，直接给你落地的实现方案：

核心思路 #

先通过正则表达式从消息文本中提取所有合法的数学元素（数字、运算符、括号），拼接成可计算的表达式，再安全执行运算并返回结果。

方案一：基础运算（安全过滤版） #

适合只需要加减乘除、括号运算的场景，用正则提取+安全过滤的eval实现，避免恶意代码执行：

import re

def extract_and_calculate(message_content):
    # 匹配整数、浮点数、正负号、基础运算符和括号
    # 支持负数（比如-123）、小数（比如3.14、.5）
    pattern = r'(-?\d+\.?\d*|\.?\d+|\+|\-|\*|\/|\(|\))'
    # 提取所有匹配的数学元素
    math_elements = re.findall(pattern, message_content)
    if not math_elements:
        return None  # 无有效运算内容
    
    # 拼接成完整表达式
    expression = ''.join(math_elements)
    
    # 严格安全检查：只允许数字、运算符、括号和小数点
    allowed_chars = set('0123456789.+-*/()')
    if not all(char in allowed_chars for char in expression):
        return None
    
    # 执行计算并处理结果
    try:
        result = eval(expression)
        # 整数结果返回int类型，否则返回float
        return int(result) if isinstance(result, float) and result.is_integer() else result
    except (SyntaxError, ZeroDivisionError):
        return None  # 表达式非法或除零错误

测试示例 #

• 输入："27+3 whats good dude" → 提取"27+3" → 返回30
• 输入："594 what's up man" → 提取"594" → 返回594
• 输入："300+73 hello" → 返回373；输入"600+13 test" → 返回613

方案二：复杂运算（更安全） #

如果需要支持次方、三角函数等复杂运算，或者想彻底避免eval的安全风险，用sympy库（专门的数学符号计算库）：

先安装依赖：

pip install sympy

实现代码：

import re
from sympy import sympify

def extract_and_calculate(message_content):
    pattern = r'(-?\d+\.?\d*|\.?\d+|\+|\-|\*|\/|\^|\(|\)|sin|cos|tan)'
    math_elements = re.findall(pattern, message_content, re.IGNORECASE)
    if not math_elements:
        return None
    
    expression = ''.join(math_elements)
    try:
        # sympify只解析数学表达式，不会执行任意代码
        result = sympify(expression)
        numeric_result = float(result)
        return int(numeric_result) if numeric_result.is_integer() else numeric_result
    except (SyntaxError, ValueError, ZeroDivisionError):
        return None

集成到Discord机器人 #

在消息事件中调用上述函数即可，示例：

import discord
from discord.ext import commands

bot = commands.Bot(command_prefix='!', intents=discord.Intents.all())

@bot.event
async def on_message(message):
    if message.author == bot.user:
        return
    # 假设用!calc命令触发计算
    if message.content.startswith('!calc'):
        content = message.content[5:]  # 去掉!calc前缀
        result = extract_and_calculate(content)
        if result is not None:
            await message.channel.send(f"计算结果：{result}")
        else:
            await message.channel.send("没找到有效的数学表达式，请检查输入~")
    await bot.process_commands(message)

bot.run('你的机器人Token')

注意事项 #

1. 正则表达式可根据需求调整，比如添加对次方^、根号等符号的支持
2. 必须处理异常情况，避免机器人崩溃
3. 如果是公开机器人，建议限制计算复杂度，防止恶意用户发送大量复杂表达式占用资源

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