pandas对某一列数据进行处理

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

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

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

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

# 此处示例的变量为一个<class 'pandas.core.frame.DataFrame'>
# 只想处理其中一列时，使用variable['paramName']来指定需要处理的列

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

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