自定义pandas函数,能够提取某一列中的数字
时间: 2024-05-12 11:17:33 浏览: 66
对pandas中Series的map函数详解
答:当然可以,你可以使用 pandas 中的 Series.str.extract() 方法来提取某一列中的数字。具体实现可以参考以下代码:
```
import pandas as pd
def extract_numbers(s):
"""从字符串中提取数字"""
numbers = pd.Series(s).str.extract('(\d+)', expand=False)
return pd.to_numeric(numbers)
# 举个例子
df = pd.DataFrame({'text': ['Hello123World', 'Python666', '99RedBalloons']})
df['numbers'] = extract_numbers(df['text'])
print(df)
```
这段代码会从 `text` 列中提取数字,并将结果存储到 `numbers` 列中。输出结果如下:
```
text numbers
0 Hello123World 123
1 Python666 666
2 99RedBalloons 99
```
阅读全文