Я только изучаю python, и я не знаю, с чего начать и go с моим текущим проектом. Я получил проект, чтобы сделать матрицу, которая может показать связь между цветом из CSV-файла в другой CSV-файл. Входной и выходной файл будут выглядеть так:
Вход
black blue
red blue
blue black
yellow black
Выход
black blue red yellow
black 0 1 0 1
blue 1 0 1 0
red 0 1 0 0
yellow 1 0 0 0
?```
Используя этот код, но .csv вывод ничего не показывает.
import csv
import collections
import itertools
grid = collections.Counter()
with open("color.csv", "r", newline="") as fp:
reader = csv.reader(fp)
for line in reader:
line = [name.strip() for name in line if name.strip()]
if len(line) == 1:
grid[line[0], line[0]] += 1
relations = sorted(set(pair[0] for pair in grid))
with open("color_sorted.csv", "w", newline="") as fp:
writer = csv.writer(fp)
writer.writerow([''] + relations)
for relation in relations:
line = [relation,] + [grid[relation, other] for other in relations]
writer.writerow(line)