python - 数据框检查字符串是否以数字结尾并删除

Question

假设我在数据框 df 中有一个 foo 列，它看起来像:

0            abc1
1             def
2           g3sse1
3           f32asd

我不想要末尾的数字，如果有的话。

0             abc
1             def
2            g3sse
3           f32asd

像这样。

我能做的最好的是:

df.foo[df['foo'].str[-1].str.isdigit()] = df['foo'].str[:-1]

这解决了问题，但是......我很好奇是否有更优雅的方法来做到这一点。我想正则表达式不会让它看起来更好，但我很欣赏任何想法!

Answer 1

由于您的输入仅包含尾随数字，在这种情况下您不想使用正则表达式，您也可以使用 rstrip和 python 的string模块:

import string
df['foo_refined'] = df['foo'].str.rstrip(string.digits)

      foo foo_refined
0    abc1         abc
1     def         def
2  g3sse1       g3sse
3  f32asd      f32asd

---

a = '12a'
>>> a.rstrip(string.digits)
'12a'

b = '12a2'
>>> b.rstrip(string.digits)
'12a'

c = '12a12x'
>>> c.rstrip(string.digits)
'12a12x'

d = '123'
>>> d.rstrip(string.digits)
''

以及对 lstrip 的引用，如果在此上下文中使用，它会按预期从开头而不是结尾去除任何数字。