我做的python常用的小技巧

时间:2023-03-08 16:32:55

在python编码过程中,总会遇到各种各样的小问题,我想着记录下来,以备查用,总结过去,是为了更好的思考与进步。

一. 去除变量中(标题中)多余的字符

数据处理过程中,遇到这样的情况:

y=['月份', 'X2', '月份.1', 'X2.1', '月份.2', 'X2.2', '月份.3', 'X2.3']

1.可以用切片的方法

y_new=[]
for i in y:
print(i[:2])
y_new.append(i[:2])

print(y_new)

我做的python常用的小技巧

2.可以用strip方法

x='X2.1'
x.strip('.1')

我做的python常用的小技巧

也可以用rstrip,lstrip;分别去除左、右的字符。

3.正则表达式

x='X2.1'

re.sub('[/.\d]','',x)

我做的python常用的小技巧

若是在处理dataframe中遇到呢?把标题中的多余的字符去掉,可以用map函数,比如:

data.columns=map(lambda x:x[:2],data.columns)

我做的python常用的小技巧

当然,也可以正则去去除,也可以用replace:str.replace(old, new[, max]) ,总之,方法很灵活,思想很灵活。

二. 解决to_csv 乱码问题

data.to_csv('XXXXX.csv',index=False,encoding='utf_8_sig') 加上encoding='utf_8_sig'就行,index=False表示没有index列,很方便实用。