Python函数变量提取问题：带单位输入转数值参与运算

嘿，我来帮你捋捋这个问题～

你现在遇到的核心问题其实是两个点：一是Python字符串的不可变性导致函数修改后的值没同步回原变量，二是当前的replaceunit函数没有返回可直接运算的数值类型。咱们一步步来解决：

1. 先解决变量赋值的核心问题 #

你调用replaceunit(T)的时候，只是执行了函数内部的操作，但没有把函数返回的新值重新赋值给T/B。因为Python里字符串是不可变对象，函数里修改的只是内部的临时副本，原变量不会自动更新。

要改成这样：

# 先判断输入不为空，避免索引报错
if T and T[-1].isalpha():
    T = replaceunit(T)  # 把函数返回的新值存回T
if B and B[-1].isalpha():
    B = replaceunit(B)  # 同理处理B

2. 完善replaceunit函数 #

原来的函数只是把单位字符替换成数字字符串，比如把6M变成61000000，但这还是字符串，没法直接代入公式运算。咱们要改成直接返回数值类型，同时加个容错判断：

def replaceunit(body):
    unit = body[-1]
    # 先检查单位是否在咱们的字典里
    if unit in units:
        # 提取数字部分转成float，再乘以单位对应的倍率
        num = float(body[:-1])
        return num * units[unit]
    else:
        # 如果没有单位或者单位不识别，直接转成数值返回
        return float(body)

3. 完整可运行的示例代码 #

把所有部分整合起来，这样处理后的T和B就是可以直接运算的数值了：

# 定义单位字典
units = {
    'M': 1e6,  # mega
    'G': 1e9   # giga
}

def replaceunit(body):
    unit = body[-1]
    if unit in units:
        num = float(body[:-1])
        return num * units[unit]
    else:
        return float(body)

# 获取用户输入
T = input("T = ")
B = input("B = ")

# 处理单位转换
if T and T[-1].isalpha():
    T = replaceunit(T)
else:
    T = float(T)
if B and B[-1].isalpha():
    B = replaceunit(B)
else:
    B = float(B)

# 代入自定义公式运算（这里用T*B做示例，你换成自己的公式就行）
result = T * B
print(f"运算结果：{result}")

额外小提示 #

• 这个代码已经支持带小数点的输入，比如6.5M也能正确转换。
• 如果担心用户输入格式错误（比如空输入、乱码），可以加个try-except块捕获异常，让程序更健壮。

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