Как создать матрицу CSV на основе файла CSV? - PullRequest
0 голосов
/ 12 марта 2020

Я только изучаю 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)
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...