python - Pandas 根据另一列的子字符串中的数字从子字符串切片中创建新列

Question

我有一个名为“表”的数据框，如下所示:

import pandas as pd
import numpy as np
table = pd.read_csv(main_data, sep='\t')

它产生这个:

NAME    SYMBOL    STRING
A       blah       A34SA
B       foo        BS2812D
...

如何在 pandas 中创建一个新列，以便我拥有以下内容:

NAME     SYMBOL      STRING    NUMBER
   A       blah       A34SA        34 
   B        foo     BS2812D      2812

到目前为止，我有这个:table['NUMBER'] = table.STRING.str[int(filter(str.isdigit, table.STRING))] 但此函数在此上下文中不起作用。

谢谢!

Answer 1

以下应该有效

table['NUMBER'] = table.STRING.apply(lambda x: int(''.join(filter(str.isdigit, x))))