Как дублировать строки CSV row_number раз?

Question

Я пытаюсь использовать Python для генерации CSV файл «расстояния», где я хочу, чтобы дублировать строки другого файла, «результат», как это:

Result.csv:

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

расстояния. CSV

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

Вот мой код Python:

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]) 

По какой-то причине он только дублирует первый ряд в result.csv

Заранее спасибо!

Answer 1

Вы можете использовать itertools.product, чтобы повторы на колонке ID в distance.csv:

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)) 

Если вам нужно больше, чем 2 дупликации: 'ID1', 'ID2',..., 'IDn', измените значение повтора в product до n

Answer 2

Поскольку csv.reader является итератором над файлом, вы можете пройти его только один раз. Попробуйте прочитать строки в списке:

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)