Pandas提取字符串列首位数字并转为数值型 - system_version列示例

{"title":"Pandas提取字符串列首位数字并转为数值型 - system_version列示例","description":"本文介绍如何使用Pandas的apply函数，结合正则表达式，将包含版本号信息的字符串列（如system_version）的首位数字提取出来，并将其转换为数值型。以'6'、'6.2.3'、'9.1'等示例值进行说明，并提供代码示例。","keywords":"pandas, apply, 正则表达式, 字符串提取, 数值转换, system_version, 版本号, 数据处理","content":"使用pandas的apply函数对system_version列进行处理，将每个值转为字符串，然后使用正则表达式提取第一个数字，最后将提取的数字转为数值型。\n\n具体代码如下：\n\npython\nimport pandas as pd\nimport re\n\n# 创建一个示例数据帧\ndata = {'system_version': ['6', '6.2.3', '9.1']}\ndf = pd.DataFrame(data)\n\n# 定义一个函数，用于提取第一个数字\ndef extract_first_number(version):\n version_str = str(version)\n first_number = re.findall(r'\d+', version_str)[0]\n return int(first_number)\n\n# 使用apply函数对system_version列进行处理\ndf['system_version'] = df['system_version'].apply(extract_first_number)\n\nprint(df)\n\n\n运行结果如下：\n\n\n system_version\n0 6\n1 6\n2 9\n\n\n可以看到，system_version列的值已经被转为了数值型，并且只保留了第一个数字。"}