pandas对某一列数据进行处理

今天遇到一个问题,小记录一下处理过程,现在要将一个pandas变量的某个字段进行转换。

之前学的是用applymap对所有字段进行批量处理,然后搜索了一些资料后,做了如下总结,使用apply方法只改某个需要的字段。

# 定义处理方法,至少要有一个入参,然后有返回内容
# 此处示例是通过判断一个字符串里面是否含有某个字,然后转换为1或0

def processMethod(d:str)->int:
    # print(d)
    if d.find('something')>0:
        return 1
    else:
        return 0

# 此处示例的变量为一个
# 只想处理其中一列时,使用variable['paramName']来指定需要处理的列

# 只处理某列时可以用apply方法,然后传入一个lambda,里面用x['paramName']再指定要传入的参数
# 必须加axis=1,否则会报错,把rooms变量的第一个字段,而不是指定的字段给传进去
# axis=1,表示每次取一行数据进行处理,按行处理

rooms['direction']=rooms.apply(lambda x:processMethod(x['direction']),axis=1)

你可能感兴趣的:(pandas,python)