將列表(list)的資料寫到csv 裡面+ 讀取csv檔案裡面的資料並寫到列表(list)裡面
阿新 • • 發佈:2018-11-27
將列表(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])