python - 如何在 Pandas 数据框中的列开头替换非整数值

Question

我有一个数据框，

 df=pd.DataFrame({"A":[1,2],"B":["a 1",np.nan]})

    A   B
0   1   a 1
1   2   NaN

我正在尝试将 df.B 值替换为其中的整数值。(即)如果我们有 abcd.. 1 我想将它转换为 1。

我试过了，

df.B.replace(np.nan,"").astype(str).str.replace('^[^\d]*', '').astype(int)

但它不转换空字符串。我正进入(状态ValueError: invalid literal for int() with base 10: '' 请帮助实现我的输出。提前致谢。

我的预期输出是，

     A   B
 0   1   1
 1   2   ''

Answer 1

如果想要提取 integer 并为所有其他值获取相同的值，请使用带有 try-except 的自定义函数:

def f(x):
    try:
        return int(x)
    except ValueError:
        return x

df.B = df.B.replace('^[^\d]*', '', regex=True).fillna('').apply(f)
print (df)
   A  B
0  1  1
1  2  

print (df.B.apply(type))
0    <class 'int'>
1    <class 'str'>
Name: B, dtype: object