如何從新創建的csv文件列中添加？

Question

我想實時創建一個文件，並將與列對應的值實時添加到相應CSV文件中的現有文件中。

如何添加在該程序中生成的每個CSV文件？

我將寫下現在使用的代碼。

import csv

for i in range(10):
SD="Save datas(Angle)"+str(i)  ## 해당 각도별로 배열을 지정

SDArray1=str(SD)               ## 파일을 만들어준다

f=open(SDArray1+".csv","a+t")#  ## 이름을 만들어준 파일을 생성

csv_writer = csv.writer(f)
csv_writer.writerow([SD])
print("One loop has started")
f.close()#

for i in range(1,5):
    cdata=[i]
    f=open(SDArray1+".csv","a+t")

    csv_writer =csv.writer(f)
    csv_writer.writerow(cdata)

    print(cdata)
    f.close()#
    print("loop's finished!")

如果您看上面的代碼，則會創建一個特定的文件。 我完成了下一個文件，但是我想知道如何在文件中添加列。

Answer 1

csv.write_row()占據一csv.write_row()列的列-如果需要更多列，請將其添加到cdata=[i] -fe cdata=[i,i*2,i*3,i*4] 。

您應該將with open() as f:來進行文件操作 ，它具有更強的抗錯誤能力，並在離開with -block時自動關閉文件。

固定：

import csv

# do not use i here and down below, thats confusing, better names are a plus
for fileCount in range(10):
    filename = "filename{}.csv".format(fileCount) # creates filename0.csv ... filename9.csv 

    with open(filename,"w") as f:#  # create file new
        csv_writer = csv.writer(f)
        # write headers
        csv_writer.writerow(["data1","data2","data3"])
        # write 4 rows of data
        for i in range(1,5):
            cdata=[(fileCount*100000+i*1000+k) for k in range(3)] # create 3 datapoints
            # write one row of data [1000,1001,1002] up to [9004000,9004001,9004002] 
            # for last i and fileCount
            csv_writer.writerow(cdata)

# no file.close- leaving wiht open() scope autocloses

檢查我們寫的內容：

import os
for d in sorted(os.listdir("./")):
    if d.endswith("csv"): 
        print(d,":") 
        print("*"*(len(d)+2)) 
        with open(d,"r") as f: 
            print(f.read()) 
        print("")

輸出：

filename0.csv :
***************
data1,data2,data3
1000,1001,1002
2000,2001,2002
3000,3001,3002
4000,4001,4002


filename1.csv :
***************
data1,data2,data3
101000,101001,101002
102000,102001,102002
103000,103001,103002
104000,104001,104002

filename2.csv :
***************
data1,data2,data3
201000,201001,201002
[...snip the rest - you get the idea ...]     

filename9.csv :
***************
data1,data2,data3
901000,901001,901002
902000,902001,902002
903000,903001,903002
904000,904001,904002

---

要將新列添加到現有文件：

• 打開舊文件進行讀取
• 打開新文件進行寫入
• 讀取舊文件頭，添加新列頭並將其寫入新文件
• 讀取所有行，向每行添加新的列值，並將其寫入新文件

例：

將列值的總和添加到文件中並作為新文件寫入：

filename = "filename0.csv"
newfile =  "filename0new.csv"

# open one file to read, open other (new one) to write
with open(filename,"r") as r, open(newfile,"w") as w:
    reader = csv.reader(r)
    writer = csv.writer(w)

    newHeader = next(reader)   # read the header
    newHeader.append("Sum")    # append new column-header
    writer.writerow(newHeader) # write header

    # for each row: 
    for row in reader:
        row.append(sum(map(int,row)))   # read it, sum the converted int values
        writer.writerow(row)            # write it

# output the newly created file:
with open(newfile,"r") as n:
    print(n.read())

輸出：

data1,data2,data3,Sum
1000,1001,1002,3003
2000,2001,2002,6003
3000,3001,3002,9003
4000,4001,4002,12003