将列表(list)的数据写到csv 里面
import pandas as pd
file_path = 'file_path'
image_id = [397133, 37777, 252219, 87038]
name=['imageid']
test=pd.DataFrame(columns=name,data=image_id)
print(test)
test.to_csv(file_path+'/'+'test.csv',index=False)
读取csv文件里面的数据并写到列表(list)里面
# -*- coding utf-8 -*-
import pandas as pd
import numpy as np
import os
import json
# image_id csv 文件的路径
file_path = '所在文件夹路径'
file_name = 'csv文件名字'
os.chdir(file_path) # 切换到csv所在路径,方便读取数据
# 将csv文件数据转化为list
data = pd.read_csv(file_name)
list = data.values.tolist()
image_id = [] # 存储的是要提取的数据
for i in range(len(list)):
image_id.append(list[i][0])