1. 程式人生 > >將列表(list)的資料寫到csv 裡面+ 讀取csv檔案裡面的資料並寫到列表(list)裡面

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