Wie kopiere ich CSV-Zeilen row_number mal?

Question

Ich versuche, Python zu verwenden, um eine CSV-Datei „Wege“ zu erzeugen, wo ich die Zeilen einer anderen Datei kopieren wollen, „Ergebnis“ wie folgt aus:

Result.csv:

|ID | 
|---| 
| 1 | 
| 2 | 
| 3 | 
... 

Entfernungen.

csv

|ID1 |ID2 | 
|----|----| 
| 1 | 1 | 
| 1 | 2 | 
| 1 | 3 | 
| 2 | 1 | 
| 2 | 2 | 
| 2 | 3 | 
... 

Hier ist mein Python-Code:

with open('distances.csv', 'wb') as ff: 
    writer = csv.writer(ff, delimiter=";", quoting=csv.QUOTE_ALL) 

    with open('result.csv', 'rb') as f: 
     reader = csv.reader(f, delimiter=";", quoting=csv.QUOTE_NONE) 
     reader2 = csv.reader(f, delimiter=";", quoting=csv.QUOTE_NONE) 

     for row in reader: 
      for row2 in reader2: 
       lst = [] 
       lst.append(row[0]) 
       lst.append(row[1]) 
       lst.append(row2[0]) 
       lst.append(row2[1]) 

       writer.writerow([unicode(s).encode("utf-8") for s in lst]) 

Aus irgendeinem Grund dupliziert es nur die erste Zeile in result.csv

Vielen Dank im Voraus!

Answer 1

Sie itertools.product können die Wiederholungen auf Spalte ID in distance.csv zu machen:

from itertools import product 
import csv 


with open('result.csv', 'r') as fin, open('distance.csv', 'w') as fout: 
    reader = csv.reader(fin, delimiter=";", quoting=csv.QUOTE_NONE) 
    writer = csv.writer(fout, delimiter=";", quoting=csv.QUOTE_ALL) 

    # skip header 
    next(reader) 

    buffer_ = [row[0] for row in reader] 

    writer.writerow(['ID1', 'ID2']) 
    writer.writerows(i for i in product(buffer_, repeat=2)) 

Wenn Sie mehr als 2 Vervielfältigungen müssen: 'ID1', 'ID2',..., 'IDn', ändern Sie den Wiederholungswert in product zu n

Answer 2

Da die csv.reader ein Iterator über die Datei ist, können Sie nur einmal durchlaufen. Versuchen Sie, die Zeilen in eine Liste zu lesen:

with open("result.csv", "b") as f, open("distances.csv", "wb") as ff: 
    reader = csv.reader(f, delimiter=";", quoting=csv.QUOTE_NONE) 
    writer = csv.writer(ff, delimiter=";", quoting=csv.QUOTE_ALL) 
    rows = list(reader) 

    for row1 in rows: 
     for row2 in rows: 
      lst = row1 + row2 
      writer.writerow(lst)