处理CSV文件,根据一个列表的内容来匹配第二列的内容,如果该列表有B列的元素,则在该元素对应的行数增加一列,新加的列输出为1
CSV数据为:
…
该数据和列表lis匹配,这个名为lis列表包括TDSC和IEEE Transactions on Mobile Computing等元素,如果CSV数据中的B列和列表中的数据一致,就在对应的新加的列数中输出1,理想效果如下:
…
请大神给出代码
回答
使用apply方法
df[‘C’] = df[‘B’].apply(lambda x: “1” if x in lis else ”)
例子如下:
In [1]: import pandas as pd
In [2]: df = pd.DataFrame({"name":["Jack", "Tony", "Robin"],"company":["Alibaba"
...: ,"Tencent",