python 逐行读取csv文件,修改内容后,逐行存入新的csv文件
程序员文章站
2024-03-22 13:30:34
...
程序实现之前的csv内容
第一种方法
import glob
import csv
#在对应的文件夹下,读取csv文件
csv_list = glob.glob('./14 points/*') #查看同文件夹下的csv文件数
#每个类别一个label值
label=[]
#用来分配label值
index=0
#1.csv 为需要写入新数据的文件,a表示追加内容,不加newline='',会出现空行
out = open('1.csv','a', newline='')
csv_write=csv.writer(out)
for i in csv_list:
#打开./14 points/ 文件中的一个csv文件,
csv_file = open(i)
#读取csv_file文件中的内容
csv_reader_lines = csv.reader(csv_file)
for one_line in csv_reader_lines:
#将类别index存入label[]
label.append(index)
#列表与列表相加,逐行写入
csv_write.writerow(label+one_line)
#列表弹出
label.pop(0)
#访问完一个类别的csv之后,index+1,
index+=1
程序实现之后的csv,文件
附加:在每行的末尾添加内容
import pandas as pd
import os
import csv
inputfile_dir="motion_RAF"
outputfile="motion_RAF.csv"
i=0
for inputfile in os.listdir(inputfile_dir):
with open(inputfile_dir + '/' + inputfile) as csvfile:
rows = csv.reader(csvfile)
with open(outputfile, 'a+', newline='') as f:
csv_write = csv.writer(f)
for row in rows:
row.append(i)
csv_write.writerow(row)
i=i+1
DataFrame=pd.read_csv(outputfile)
print(DataFrame)