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)
# 結果：

寫CSV

stu1 = [lid, k, pre_count_data[k]]
# 打開文件，寫模式為追加'a'
out = open('../results/write_file.csv', 'a', newline='')
# 設定寫入模式
csv_write = csv.writer(out, dialect='excel')
# 寫入具體內容
csv_write.writerow(stu1)

到此這篇關于Pandas保存csv數(shù)據(jù)的三種方式詳解的文章就介紹到這了,更多相關Pandas保存csv內容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: