Python中从字符串提取整数及转换带百分号NumPy数组为数值数组的方法
问题解答
一、从字符串中拆分出整数
处理这类场景最顺手的工具是正则表达式,Python的re模块能轻松搞定不同格式的整数提取需求,比如字符串混着字母、符号的情况,甚至带正负号的整数也能一并处理。
举个实际例子,假设你有字符串"今天赚了120块,昨天亏了-30块,累计结余90块",提取里面的整数可以这么写:
import re s = "今天赚了120块,昨天亏了-30块,累计结余90块" integers = list(map(int, re.findall(r'-?\d+', s))) print(integers) # 输出: [120, -30, 90]
简单解释下:
- 正则表达式
r'-?\d+'里,-?匹配可选的负号,\d+匹配一个或多个连续数字 re.findall会返回所有匹配到的字符串结果,再通过map(int, ...)把它们转成整数类型
如果你的字符串里只有单个整数(比如"本次测试得分:87"),直接取结果列表的第一个元素就可以了。
二、将带百分号的NumPy数组转换为数值数组
针对你给出的数组A,我们需要去掉每个元素末尾的%符号,得到纯数值的数组。这里有两种实用的实现方式:
方法1:NumPy内置字符串函数(效率最高)
import numpy as np # 初始化你的目标数组 A = np.array(['83.56%', '2.74%', '2.74%', '4.11%', '4.11%', '19.18%', '76.71%', '20.55%', '34.25%', '54.79%']) # 去掉每个元素末尾的%,得到字符串类型的数值数组 B_str = np.char.rstrip(A, '%') print(B_str) # 输出: ['83.56' '2.74' '2.74' '4.11' '4.11' '19.18' '76.71' '20.55' '34.25' '54.79'] # 如果需要转成可计算的float类型数组(更实用) B_float = B_str.astype(float) print(B_float) # 输出: [83.56 2.74 2.74 4.11 4.11 19.18 76.71 20.55 34.25 54.79]
方法2:列表推导式(更直观,适合新手理解)
如果你对NumPy的字符串操作不太熟悉,用列表推导式逐个处理元素也很清晰:
import numpy as np A = np.array(['83.56%', '2.74%', '2.74%', '4.11%', '4.11%', '19.18%', '76.71%', '20.55%', '34.25%', '54.79%']) # 截取每个元素除最后一位(%)的部分,再转成数组 B_str = np.array([item[:-1] for item in A]) # 要是直接要float数组,就写成:B_float = np.array([float(item[:-1]) for item in A]) print(B_str)
两种方法都能满足需求,推荐用方法1,因为NumPy的内置函数处理大规模数组时效率会更高。
内容的提问来源于stack exchange,提问作者Soumav Nath
相关产品推荐
相关产品推荐

