我正在尝试将csv文件中的多行组合在一起。我可以在Excel中轻松完成此操作,但是我想对数百个文件执行此操作,因此我需要将其作为代码。我试图将行存储在数组中,但似乎不起作用。我正在用Python做到这一点。
假设我有一个csv文件;
1,2,3
4,5,6
7,8,9
我要做的就是拥有一个csv文件;
1,2,3,4,5,6,7,8,9
我尝试过的代码是这样;
fin = open("C:\\1.csv", 'r+')
fout = open("C:\\2.csv",'w')
for line in fin.xreadlines():
new = line.replace(',', ' ', 1)
fout.write (new)
fin.close()
fout.close()
能否请你帮忙?
最佳答案
您应该为此使用csv
模块,因为在逗号上手动拆分CSV非常容易出错(单列可以包含带逗号的字符串,但最终会错误地将其拆分为多列)。 CSV模块使用值列表来表示单行。
import csv
def return_contents(file_name):
with open(file_name) as infile:
reader = csv.reader(infile)
return list(reader)
data1 = return_contents('csv1.csv')
data2 = return_contents('csv2.csv')
print(data1)
print(data2)
combined = []
for row in data1:
combined.extend(row)
for row in data2:
combined.extend(row)
with open('csv_out.csv', 'w', newline='') as outfile:
writer = csv.writer(outfile)
writer.writerow(combined)
该代码为您提供了该方法的基础,但是将其扩展到数百个文件将是很丑陋的。相反,您可能希望
os.listdir
将所有文件一个一个地拉到一个目录中,然后将它们添加到输出中。这就是我将阅读代码打包到return_contents
函数中的原因。我们只需使用一组代码即可对不同的文件重复相同的过程数百万次,以进行实际的读取。像这样:import csv
import os
def return_contents(file_name):
with open(file_name) as infile:
reader = csv.reader(infile)
return list(reader)
all_files = os.listdir('my_csvs')
combined_output = []
for file in all_files:
data = return_contents('my_csvs/{}'.format(file))
for row in data:
combined_output.extend(row)
with open('csv_out.csv', 'w', newline='') as outfile:
writer = csv.writer(outfile)
writer.writerow(combined_output)
关于python - 使用Python将两行合并为一个csv文件,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/53670209/