从通用边列表在 python 中创建边列表

Question

我有一个 csv，其中每一行都是一个文档。每行中的第一个值是 ID。每行中的所有后续值（第一个值之后）都是文档中引用的文档。

例如，在我的 csv 中，假设我有四个文档（即四行）：

A, B, D, E, G, H
B, C, E
C, I, K, L
D, A, C, H, I

因此，对于文件 1 (A)，该文件引用了其他五个文件（B、D、E、G、H）。

我想创建一个边缘列表（并导出到新的 csv），以便：

A, B
A, D
A, E
A, G
A, H
B, C
B, E
C, I
C, K
C, L
D, A
D, C
D, H
D, I

如何使用 python 创建新的边缘列表 csv 文件？

感谢您的帮助。

Answer 1

您可以只使用几个 for 循环

l = [['A', 'B', 'D', 'E', 'G', 'H'],
    ['B', 'C', 'E'],
    ['C', 'I', 'K', 'L'],
    ['D', 'A', 'C', 'H', 'I']]
edge = []            
for i in l:
    for j in i[1:]:
        edge.append([i[0],j])
for i in edge:
    print i

或者更简单的是使用列表理解

l = [['A', 'B', 'D', 'E', 'G', 'H'],
    ['B', 'C', 'E'],
    ['C', 'I', 'K', 'L'],
    ['D', 'A', 'C', 'H', 'I']]   

edge = [[i[0],j] for i in l for j in i[1:]]

for i in edge:
    print i

['A', 'B']
['A', 'D']
['A', 'E']
['A', 'G']
['A', 'H']
['B', 'C']
['B', 'E']
['C', 'I']
['C', 'K']
['C', 'L']
['D', 'A']
['D', 'C']
['D', 'H']
['D', 'I']