解析後寫入文件

Question

我寫了一個小型python代碼，它將讀取一個示例csv文件並將其第一列複製到一個臨時csv文件。現在，當我嘗試將該臨時文件與另一個文本文件進行比較並嘗試將結果寫入另一個稱爲結果文件的文件時，該文件將被創建但空內容。

但是，當我在塊測試它，它工作正常

import csv 
f = open("sample.csv", "r") 
reader = csv.reader(f) 

data = open("temp1.csv", "w") 
w = csv.writer(data) 
for row in reader: 
    my_row = [] 
    my_row.append(row[0]) 
    w.writerow(my_row) 

with open('temp1.csv', 'r') as file1: 
    with open('serialNumber.txt', 'r') as file2: 
     same = set(file1).intersection(file2) 

with open('result.txt', 'w') as file_out: 
    for line in same: 
     file_out.write(line) 
     print line 

sample.csv

M11435TDS144，STB＃1，路由器＃1

M11543TH4292，STB＃2 ，Router＃1

M11509TD9937，STB＃3，Router＃1

M11543TH4258，STB＃4，路由器＃1

serialNumber.txt

G1A114042400571

M11543TH4258

M11251TH1230

M11435TDS144

M11543TH4292

M11509TD9937

Answer 1

你應該關閉輸出文件（temp1.csv），你可以從中讀取數據之前。

import csv 
f = open("sample.csv", "r") 
reader = csv.reader(f) 

data = open("temp1.csv", "w") 
w = csv.writer(data) 
for row in reader: 
    my_row = [] 
    my_row.append(row[0]) 
    w.writerow(my_row) 
data.close() # <--- Should close it before reading it in the same program !! 

with open('temp1.csv', 'r') as file1: 
    with open('serialNumber.txt', 'r') as file2: 
     same = set(file1).intersection(file2) 

with open('result.txt', 'w') as file_out: 
    for line in same: 
     file_out.write(line) 
     print line 

Answer 2

關於代碼點：

• data文件句柄沒有關閉。在寫入temp1.csv後data.close()。
• 在您的代碼same = set(file1).intersection(file2)中，您直接將文件句柄file2傳遞到交點。它期望列表。這確切的問題是。它應該是same = set(file1.readlines()).intersection(file2.readlines())

工作代碼：temp1.csv的

import csv 
f = open("sample.csv", "r") 
reader = csv.reader(f) 

data = open("temp1.csv", "wb") 
w = csv.writer(data) 

for row in reader: 
    my_row = [] 
    if len(row) != 0: 
     my_row.append(row[0]) 
     w.writerow(my_row) 
#File should be closed 
data.close() 

with open('temp1.csv', 'r') as file1: 
    with open('serialNumber.txt', 'r') as file2: 
     tmp_list = (file1.readlines()) 
     ser_list = (file2.readlines()) 
     same = set(file1.readlines()).intersection(file2.readlines()) 


with open('result.txt', 'w') as file_out: 
    for line in same: 
     file_out.write(line) 

內容：的Result.txt的

M11435TDS144 
M11543TH4292 
M11509TD9937 
M11543TH4258 

內容：

M11543TH4258 
M11543TH4292 
M11435TDS144 

您可以使用with打開文件sample.csv和temp1。csv如下。

import csv 

with open("sample.csv") as f: 
    with open("temp1.csv",'wb') as data: 
    reader = csv.reader(f) 
    w = csv.writer(data) 
    for row in reader: 
     my_row = [] 
     my_row.append(row[0]) 
     w.writerow(my_row) 


with open('temp1.csv', 'r') as file1: 
    with open('serialNumber.txt', 'r') as file2: 
     same = set(file1.readlines()).intersection(file2.readlines()) 

with open('result.txt', 'w') as file_out: 
    for line in same: 
     file_out.write(line)