日本免费高清视频-国产福利视频导航-黄色在线播放国产-天天操天天操天天操天天操|www.shdianci.com

學(xué)無(wú)先后,達(dá)者為師

網(wǎng)站首頁(yè) 編程語(yǔ)言 正文

Pandas保存csv數(shù)據(jù)的三種方式詳解_python

作者:AI浩 ? 更新時(shí)間: 2022-05-31 編程語(yǔ)言

方法一

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ù)的保存,還能讀寫(xiě)CSV文件,下面是實(shí)現(xiàn)的核心代碼

使用pandas讀取CSV

import pandas as pd
import csv
if name == '__main__':

# header=0——表示csv文件的第一行默認(rèn)為dataframe數(shù)據(jù)的行名稱(chēng),
# 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é)果: 

寫(xiě)CSV

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

原文鏈接:https://blog.csdn.net/hhhhhhhhhhwwwwwwwwww/article/details/123791632

欄目分類(lèi)
最近更新