python处理一个csv数据,根据其中一列数据来得到新一列写入同一个文件中?

如图所示的表格,根据gender进行判断得到一列sort的值,如果为man,则sort为1我之前自己试着写的时候直接把gender一列提出来对每一个值进行判断,但是发现不会把相应的值赋给同一行的sort了。

用pandas库可以实现:

import pandas as pd
file = 'your_csv_path.csv'
data = pd.read_csv(file)

data['sort'] = data['gender'].map(lambda x: 1 if x == 'man' else -1)
data.to_csv(file, index=False)

温馨提示:答案为网友推荐,仅供参考
第1个回答  2018-07-04