0

次のようなデータを持つ .csv があります。

30,972.74176,0.446378,1.825426,cl+l  
31,1015.075579,0.279686,2.950118,l  
32,1072.59334,0.273239,4.408458,cl+l  
33,1139.55317,-1.616904,4.756378,l  
34,1279.86304,50.335186,3.877577,r  
35,1181.541419,61.399389,5.155133,r  
36,1097.421199,63.809733,5.37508,r  
37,1018.555678,64.034432,2.59232,r

最後の列 (私のプログラムでは E と呼んでいます) が "l" または "cl+l" である行を取り出し、それを "left.csv" という別のファイルにコピーするプログラムを作成しました。 . 私がやりたいことは、2 番目の列 (B) からデータを取得する left.csv に別の列を追加し、それらの間にコンマを入れてから、列 D からのデータを追加することです。

したがって、現在の left.csv は次のようになります。

30,972.74176,0.446378,1.825426,cl+l  
31,1015.075579,0.279686,2.950118,l  
32,1072.59334,0.273239,4.408458,cl+l  
33,1139.55317,-1.616904,4.756378,l

私が欲しいのは：

30,972.74176,0.446378,1.825426,cl+l,"972.74176,1.825426"  
31,1015.075579,0.279686,2.950118,l,"1015.075579,2.950118"    
32,1072.59334,0.273239,4.408458,cl+l,"1072.59334,4.408458"   
33,1139.55317,-1.616904,4.756378,l,"1139.55317,4.756378"

これを達成するにはどうすればよいですか? また、csv であってもコンマで次のエントリを開始しないようにプログラムに指示するにはどうすればよいですか? それは可能ですか?おそらく括弧または別の区切り文字を使用する必要がありますか? 私が今持っているコードは次のとおりです。

import csv
import sys
import operator

fieldnames = ["A","B","C","D","E"]
surveyfile = open("source.csv", "r")
left_file = open("left.csv",'wb')
right_file = open("right.csv",'wb')
coil_file = open("coil1.csv","wb")


reader = csv.DictReader(surveyfile, fieldnames=fieldnames, delimiter=",")
left_writer = csv.DictWriter(left_file, fieldnames, delimiter=",")
sortedlefts = sorted(reader,key=lambda x:float(x["B"]))

surveyfile.seek(0,0)

right_writer = csv.DictWriter(right_file, fieldnames, delimiter=",")
sortedrights = sorted(reader,key=lambda x:float(x["B"]), reverse=True)

coil_writer = csv.DictWriter(coil_file, fieldnames, delimiter=",")

for row in sortedlefts:
    if row["E"] == "l" or row["E"] == "cl+l":
        left_writer.writerow(row)
        coil_writer.writerow(row)   


for row in sortedrights:
    if row["E"] == "r":
        right_writer.writerow(row)
        coil_writer.writerow(row)
4

1 に答える 1

3

に値を追加するだけrowで、カンマを含むようにフォーマットされます。

row['F'] = '%s,%s' % (row['E'], row['B'])

または使用','.join()：

row['F'] = ','.join([row['E'], row['B']])

'F'ただし、次のフィールドに含めてleft_writerください。

left_writer = csv.DictWriter(left_file, fieldnames + ['F'], delimiter=",")

余分な列を無視するcoil_writerようにインスタンスを構成する場合、そのインスタンス用に別のループを作成したり、既存のループにコピーを作成したりする必要はありません。row

coil_writer = csv.DictWriter(coil_file, fieldnames, delimiter=",", extrasaction='ignore')

csvライターが見積もりを処理します。

于 2012-12-05T16:59:05.153 に答える