给csv或txt文件加上一列id

---

前言

从这样

变成这样

代码

import pandas as pd
for i in range(0,10):
    data = pd.read_csv('/home/yin/DREAMwalk-main/DREAMwalk-main/demo/LR/result/disease_label_herb_drug_{}.txt'.format(i),sep='\t', header=0)
    n = len(data)+1
    nlist = range(1,n)
    data['id'] = nlist
    
    # data['label'] = 0
    # print(data.head())
    data.to_csv('/home/yin/DREAMwalk-main/DREAMwalk-main/demo/LR/result_id/disease_label_herb_drug_id_{}.txt'.format(i),columns=['id','drug','disease','label'],index=0,header=1,sep='\t')
    ### 使用 to_csv 方法保存文件，指定列顺序为 ['id', 'drug', 'disease', 'label']，以确保输出文件的列顺序一致

如果不是保存成文件

import pandas as pd
data = pd.DataFrame.from_dict({'A':[1,2,3],'B':["a","b","c"]})
n = len(data)+1
data['id'] = range(1,n)
##data[['id','A','B']]
data.columns = ['id','A','B']

这种data.columns = [‘id’,‘A’,‘B’]就会出现列名与内容不匹配的情况，使用

import pandas as pd
data = pd.DataFrame.from_dict({'A':[1,2,3],'B':["a","b","c"]})
n = len(data)+1
data['id'] = range(1,n)
data[['id','A','B']]

重新排列列的顺序：使用 data[[‘id’, ‘A’, ‘B’]] 重新排列列的顺序，使得 id 列在最前面