具有 lambda 的 Python 熊猫应用困难

Question

我正在运行以下函数，但不知何故努力让它考虑长度条件（if 部分）。如果只有函数，它只会运行第一部分：

stringDataFrame.apply(lambda x: x.str.replace(r'[^0-9]', '') if (len(x) >= 7) else x)

不知何故，它只运行x.str.replace(r'[^0-9]', '') 部分，我在这里做错了什么，我被卡住了。

Answer 1

当您需要单独处理每个值时，您可以使用applymap，因为apply 与all column (Series) 一起使用。

然后，不要使用str.replace，而是使用re.sub，这对正则表达式更有效：

print (stringDataFrame.applymap(lambda x: re.sub(r'[^0-9]', '', x) if (len(x) >= 7) else x))

示例：

import pandas as pd
import re

stringDataFrame = pd.DataFrame({'A':['gdgdg454dgd','147ooo2', '123ss45678'],
                                'B':['gdgdg454dgd','x142', '12345678a'],
                                'C':['gdgdg454dgd','xx142', '12567dd8']})

print (stringDataFrame)
             A            B            C
0  gdgdg454dgd  gdgdg454dgd  gdgdg454dgd
1      147ooo2         x142        xx142
2   123ss45678    12345678a     12567dd8

print (stringDataFrame.applymap(lambda x: re.sub(r'[^0-9]', '', x) if (len(x) >= 7) else x))
          A         B       C
0       454       454     454
1      1472      x142   xx142
2  12345678  12345678  125678