Pandas保存csv數(shù)據(jù)的三種方式詳解
方法一
import os import pandas as pd path = 'data/train/' img_label_list=[] testList = os.listdir(path) for file in testList: label='aa' img_label_list.append([file, label]) df1 = pd.DataFrame(data=img_label_list, columns=['id', 'label']) df1.to_csv('result.csv',index=False)
方法二
import os import pandas as pd path = 'data/train/' img_list=[] lable_list=[] testList = os.listdir(path) for file in testList: img_list.append(file) label='aa' lable_list.append(label) img_label_list2 = list(zip(img_list, lable_list)) df3 = pd.DataFrame(data=img_label_list2, columns=['filepath', 'label']) df3.to_csv('result.csv',index=False)
方法三
import os import pandas as pd path = 'data/train/' img_list=[] lable_list=[] testList = os.listdir(path) for file in testList: img_list.append(file) label='aa' lable_list.append(label) df = pd.DataFrame({"filename": img_list, "label": lable_list}) df.to_csv('result.csv',index=False)
補(bǔ)充
當(dāng)然Pandas不僅可以實(shí)現(xiàn)CSV文件數(shù)據(jù)的保存,還能讀寫CSV文件,下面是實(shí)現(xiàn)的核心代碼
使用pandas讀取CSV
import pandas as pd import csv if name == '__main__': # header=0——表示csv文件的第一行默認(rèn)為dataframe數(shù)據(jù)的行名稱, # index_col=0——表示使用第0列作為dataframe的行索引, # squeeze=True——表示如果文件只包含一列,則返回一個(gè)序列。 file_dataframe = pd.read_csv('../datasets/data_new_2/csv_file_name.csv', header=0, index_col=0, squeeze=True) # 結(jié)果:
寫CSV
stu1 = [lid, k, pre_count_data[k]] # 打開文件,寫模式為追加'a' out = open('../results/write_file.csv', 'a', newline='') # 設(shè)定寫入模式 csv_write = csv.writer(out, dialect='excel') # 寫入具體內(nèi)容 csv_write.writerow(stu1)
到此這篇關(guān)于Pandas保存csv數(shù)據(jù)的三種方式詳解的文章就介紹到這了,更多相關(guān)Pandas保存csv內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家!
相關(guān)文章
詳解Python+OpenCV進(jìn)行基礎(chǔ)的圖像操作
OpenCV是一個(gè)用于計(jì)算機(jī)視覺和圖像操作的免費(fèi)開源庫,有數(shù)千種優(yōu)化的算法和函數(shù)用于各種圖像操作。本文將使用OpenCV在Python中進(jìn)行一些圖像操作,感興趣的可以學(xué)習(xí)一下2022-02-02django執(zhí)行原生SQL查詢的實(shí)現(xiàn)
本文主要介紹了django執(zhí)行原生SQL查詢的實(shí)現(xiàn),主要有兩種方法實(shí)現(xiàn),文中通過示例代碼介紹的非常詳細(xì),具有一定的參考價(jià)值,感興趣的小伙伴們可以參考一下2021-08-08Python3.2中的字符串函數(shù)學(xué)習(xí)總結(jié)
這篇文章主要介紹了Python3.2中的字符串函數(shù)學(xué)習(xí)總結(jié),本文講解了格式化類方法、查找 & 替換類方法、拆分 & 組合類方法等內(nèi)容,需要的朋友可以參考下2015-04-04