pandas遍历数据并输出到指定csv文件中_遍历文件 寻找特定数据 并保存到新csv文件

主要来源是想将自己想要的信息统计到csv文件中

import pandas as pd
# Mass
data = pd.read_csv("filename.csv")
list = data['HLA'].unique()
pos_num = []
neg_num = []
 
# 首先创建一个df文件
result = pd.DataFrame(columns=('HLA','pos_num','neg_num','count','target','length'))
for i in range(len(list)):
    # 根据条件筛选
    allele = data[(data['HLA'].str.contains(list[i]))&(data['Peptide'].str.len() == 9)]
 
    # 想要获取的数据
    pos_num = len(allele['Target'] == 1)
    neg_num = len(allele['Target'] == 0)
    count = pos_num + neg_num
 
    # 逐条进行添加
    result = result.append(pd.DataFrame({'HLA':[list[i]],'pos_num':[pos_num],'neg_num':[neg_num],'count':[count],'target':['Mass'],'length':[9]}),ignore_index=True)
 
# 将结果输出到csv文件中，或者什么文件都可以
result.to_csv("dataset/out_result.csv",index=False,sep=',')