模块导入
import pandas as pd
import numpy as np
去除函数
def deal_space(s):
if isinstance(s, float) and np.isnan(s):
return s
elif isinstance(s, int) or isinstance(s, float):
return str(s).strip()
elif isinstance(s, str):
return s.strip()
return s
去除前后空格
def deal_blank(data, cols):
for col in cols:
data[col] = data[col].map(deal_space)
return data
函数使用
# data为DataFrame格式
res = deal_blank(data, [列名, 列名])