如果长度不等于x，则Python将字符串替换为空

Question

我有以下数据帧：

df=pd.DataFrame({'ssn':[12345,54321,111,47895,222311],'Name':['john','mike','adam','doug','liz']})

DataFrame包含一个'ssn'，它应该只包含5位数字。 我想用空格替换包含小于或大于5位的所有行。

所需的输出如下：

   Name   ssn
0  john   12345
1  mike   54321
2  adam   
3  doug   47895
4  liz    

如果长度小于x ，我从SO 替换字符串中引用以下帖子但是，使用相同的解决方案和以下命令时给出了一个错误：

mask = df['ssn'].str.len() == 5
df['ssn'] = df['ssn'].mask(mask, df['ssn'].str.replace(df['ssn'], ''))
Traceback (most recent call last): 
TypeError: 'Series' objects are mutable, thus they cannot be hashed

我将不胜感激任何建议。

Answer 1

您也可以使用df.apply执行此df.apply ： df['ssn'] = df['ssn'].apply(lambda a: a if len(str(a))==5 else '') 。

Answer 2

你的列ssn包含的数字不是字符串，这就是它无效的原因。 请尝试以下方法：

mask = df['ssn'].astype(str).str.len() != 5
df.loc[mask, 'ssn'] = ''

In [1] : print(df)
Out[1] :    Name    ssn
0  john  12345
1  mike  54321
2  adam       
3  doug  47895
4   liz