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)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家!
相關(guān)文章
詳解js文件通過python訪問數(shù)據(jù)庫方法
在本篇內(nèi)容里小編給大家分享了關(guān)于js文件通過python訪問數(shù)據(jù)庫方法和技巧,有需要的朋友們跟著學(xué)習(xí)參考下。2019-03-03
理解Django 中Call Stack機(jī)制的小Demo
這篇文章主要介紹了理解Django 中Call Stack 機(jī)制的小Demo,文中通過示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2020-09-09
python tkinter庫實(shí)現(xiàn)氣泡屏保和鎖屏
這篇文章主要為大家詳細(xì)介紹了python tkinter庫實(shí)現(xiàn)氣泡屏保和鎖屏,具有一定的參考價(jià)值,感興趣的小伙伴們可以參考一下2019-07-07
Python基于smtplib實(shí)現(xiàn)異步發(fā)送郵件服務(wù)
這篇文章主要介紹了Python基于smtplib實(shí)現(xiàn)異步發(fā)送郵件服務(wù),需要的朋友可以參考下2015-05-05
Python給Excel寫入數(shù)據(jù)的四種方法小結(jié)
本文主要介紹了Python給Excel寫入數(shù)據(jù)的四種方法小結(jié),包含openpyxl庫、xlsxwriter庫、pandas庫和win32com庫,具有一定的參考價(jià)值,感興趣的可以了解一下2025-02-02
使用Python將圖片轉(zhuǎn)正方形的兩種方法實(shí)例代碼詳解
這篇文章主要介紹了使用Python將圖片轉(zhuǎn)正方形的兩種方法,本文通過實(shí)例代碼給大家給大家介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值,需要的朋友可以參考下2020-04-04

