将列表（list）的数据写到csv 里面+ 读取csv文件里面的数据并写到列表（list）里面

将列表(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])