pandas保存csv数据的几种方式。
Posted AI浩
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了pandas保存csv数据的几种方式。相关的知识，希望对你有一定的参考价值。
第一种：
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保存csv数据的几种方式。的主要内容，如果未能解决你的问题，请参考以下文章
Pandas 基础 - 生成 Dataframe 的几种方式