首页
学习
活动
专区
工具
TVP
发布
精选内容/技术社群/优惠产品,尽在小程序
立即前往

Pandas/Python -创建新列,仅从列字符串中提取数字

在Pandas/Python中,可以使用正则表达式和字符串操作来创建新列,仅从列字符串中提取数字。

首先,我们需要确保已经导入了Pandas库:

代码语言:txt
复制
import pandas as pd

假设我们有一个名为df的数据框(DataFrame),其中包含一个名为"column_name"的列,其值是包含数字的字符串。我们希望从该列中提取出数字并创建一个新列。

下面是实现的步骤:

  1. 使用正则表达式提取数字:
  2. 使用正则表达式提取数字:
  3. 此代码行使用str.extract函数和正则表达式(\d+)从"column_name"列的每个字符串中提取一个或多个数字,并将结果存储在"new_column"列中。
  4. 如果提取的数字为多个,可以选择只保留第一个数字,可以使用.str.extract函数的expand参数设置为False,并添加.str[0],如下所示:
  5. 如果提取的数字为多个,可以选择只保留第一个数字,可以使用.str.extract函数的expand参数设置为False,并添加.str[0],如下所示:
  6. 这将只提取并存储每个字符串中的第一个数字。

完整示例代码如下所示:

代码语言:txt
复制
import pandas as pd

# 创建示例数据
data = {'column_name': ['abc123def', 'xyz456', '123']}
df = pd.DataFrame(data)

# 提取数字并创建新列
df['new_column'] = df['column_name'].str.extract('(\d+)', expand=False).str[0]

print(df)

输出结果为:

代码语言:txt
复制
  column_name new_column
0  abc123def        123
1     xyz456        456
2         123        123

在这个例子中,我们成功地从"column_name"列的每个字符串中提取了数字,并将结果存储在了"new_column"列中。

请注意,以上代码中没有提及腾讯云相关产品和链接地址,因为这与问题本身无关。如果您需要了解与云计算相关的腾讯云产品和链接,您可以查阅腾讯云官方文档或咨询腾讯云的技术支持团队。

页面内容是否对你有帮助?
有帮助
没帮助

相关·内容

  • 一场pandas与SQL的巅峰大战(二)

    上一篇文章一场pandas与SQL的巅峰大战中,我们对比了pandas与SQL常见的一些操作,我们的例子虽然是以MySQL为基础的,但换作其他的数据库软件,也一样适用。工作中除了MySQL,也经常会使用Hive SQL,相比之下,后者有更为强大和丰富的函数。本文将延续上一篇文章的风格和思路,继续对比Pandas与SQL,一方面是对上文的补充,另一方面也继续深入学习一下两种工具。方便起见,本文采用hive环境运行SQL,使用jupyter lab运行pandas。关于hive的安装和配置,我在之前的文章MacOS 下hive的安装与配置提到过,不过仅限于mac版本,供参考,如果你觉得比较困难,可以考虑使用postgreSQL,它比MySQL支持更多的函数(不过代码可能需要进行一定的改动)。而jupyter lab和jupyter notebook功能相同,界面相似,完全可以用notebook代替,我在Jupyter notebook使用技巧大全一文的最后有提到过二者的差别,感兴趣可以点击蓝字阅读。希望本文可以帮助各位读者在工作中进行pandas和Hive SQL的快速转换。本文涉及的部分hive 函数我在之前也有总结过,可以参考常用Hive函数的学习和总结。

    02
    领券