Pandas保存csv數(shù)據(jù)的三種方式詳解
更新時間:2022年03月28日 14:36:13 作者:AI浩
CSV文件以純文本形式存儲表格數(shù)據(jù)(數(shù)字和文本),是一種通用的、相對簡單的文件格式,被用戶、商業(yè)和科學廣泛應用。本文介紹了三種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)
補充
當然Pandas不僅可以實現(xiàn)CSV文件數(shù)據(jù)的保存,還能讀寫CSV文件,下面是實現(xiàn)的核心代碼
使用pandas讀取CSV
import pandas as pd import csv if name == '__main__': # header=0——表示csv文件的第一行默認為dataframe數(shù)據(jù)的行名稱, # index_col=0——表示使用第0列作為dataframe的行索引, # squeeze=True——表示如果文件只包含一列,則返回一個序列。 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)文章
Python3.2中的字符串函數(shù)學習總結(jié)
這篇文章主要介紹了Python3.2中的字符串函數(shù)學習總結(jié),本文講解了格式化類方法、查找 & 替換類方法、拆分 & 組合類方法等內(nèi)容,需要的朋友可以參考下2015-04-04