qingdan/data_validate2.py

import pandas
import json
import numpy as np
import asyncio
from subdir import db as dbservice
from subdir import service
import os
import hashlib


with open("tj_label.txt", "r") as f:
    content = f.read()
labels = content.split("\n")

def func1(input):
    for i in range(len(labels)):
        if labels[i] == input['label']:
            return str(i)
        


a = pandas.read_csv('validate1.csv', dtype='str')
a["id"] = a.apply(lambda x: func1(x), axis=1)
a.to_csv("validate2.csv")