我对Python还是很陌生,所以很抱歉,如果这很明显,但是我的代码很简单,所以我不知道它出了什么问题。我正在使用csv模块读取一个文件,其中包含来自两个不同来源(L和R)的信息,并将其拆分为两个新文件。问题是,当它创建新文件时,会将它们创建为.txt文件,这是我在csv编写中从未见过的。
import csv
with open('dataset1.csv') as csvfile:
filein = csv.reader(csvfile)
with open('dataset1Left', 'w', newline = '') as fixleft:
fixleftwriter = csv.writer(fixleft)
with open('dataset1Right', 'w', newline = '') as fixright:
fixrightwriter = csv.writer(fixright)
for row in filein:
if (row[2] == 'L'):
fixleftwriter.writerows([row])
elif (row[2] == 'R'):
fixrightwriter.writerows([row])
csvfile.close()
由于您是python的新手,因此您可能不知道pandas是处理csv文件的理想选择。要完成与您相同的操作,熊猫中的代码将是
import pandas as pd
a = pd.read_csv('dataset1.csv',header=None)
a[a[2]=='L'].to_csv('dataset1left.csv',index=False,header=False)
a[a[2]=='R'].to_csv('dataset1right.csv',index=False,header=False)
我不希望使用该解决方案的唯一原因是,如果您的文件很大,则会占用可用内存。
本文收集自互联网,转载请注明来源。
如有侵权,请联系 [email protected] 删除。
我来说两句