Python如何从'<5s'或'<0.5s'类字符串中提取浮点数？

你的问题在于原代码只提取了单个数字字符，没法识别包含小数点的完整数值。咱们换用正则表达式来匹配连续的数字和小数点组合，就能完美处理整数和小数两种情况了。

方法实现 #

使用Python的re模块来匹配字符串中的数值部分，正则表达式可以精准识别整数（如5）和小数（如0.5）格式：

import re

def get_duration_value(time_str):
    # 匹配包含整数和小数的数值：匹配一个或多个数字，可选的小数点加后续数字
    match_result = re.search(r'\d+\.?\d*', time_str)
    if match_result:
        num_str = match_result.group()
        # 自动判断返回int还是float类型
        return float(num_str) if '.' in num_str else int(num_str)
    # 如果没有匹配到数值，返回None或者根据需求自定义默认值
    return None

# 测试示例
print(get_duration_value('< 5s'))    # 输出: 5
print(get_duration_value('< 0.5s'))  # 输出: 0.5
print(get_duration_value('10.75s'))  # 输出: 10.75
print(get_duration_value('3s'))      # 输出: 3

代码说明 #

1. 正则表达式r'\d+\.?\d*'：
   • \d+：匹配一个或多个数字（保证至少有一位整数部分）
   • \.?：匹配可选的小数点（?表示该字符出现0次或1次）
   • \d*：匹配零个或多个数字（处理小数部分，允许没有小数的情况）
2. 类型判断：通过检查字符串中是否包含小数点，决定返回int还是float类型，兼顾两种场景的需求。
3. 容错处理：如果输入字符串没有匹配到数值，返回None，你也可以根据实际需求改成默认值（比如0）。

为什么原代码失效？ #

原代码用filter(str.isdigit, a)会逐个字符过滤，只保留纯数字字符。对于< 0.5s，过滤后得到的是['0', '5']，取第一个元素自然是0，丢失了小数点和完整的小数部分，所以没法正确提取带小数点的数值。

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