python - 如何从 Python 的邻接表中选择一个子矩阵？

Question

我有一个邻接列表，其中每个数组代表该行的非零列(例如，下面的 adj.list 中的第 0 个数组表示第 2 列和第 6 列为 1，其他所有列为 0)。

adj_list = [[2, 6], [1, 3, 24], [2, 4], [3, 5, 21], [4, 6, 10], [1, 5, 7], [6, 8, 9], [7], [7, 10, 14], [5, 9, 11], [10, 12, 18], [11, 13], [12, 14, 15], [9, 13], [13, 16, 17], [15], [15], [11, 19, 20], [18], [18], [4, 22, 23], [21], [21], [2, 25, 26], [24], [24]]

鉴于这个形容词。列表，我想选择一个子矩阵，它具有相同的行和列索引，由下式给出:

submatrix = (0, 1, 2, 5, 22)

子矩阵中的每个元素表示一个行号。

1) 对于子矩阵中的每一行 i，我需要从 adj_list 中获取 ith<​​ 数组(相当于获取 来自邻接矩阵的第 行)

2) 然后从那个数组中，我需要提取与子矩阵匹配的项

例如，如果我当前正在查看 submatrix 中的第 3 个元素，即 5，那么我需要转到 adj_list 中的第 5 个数组(相当于获取 adj.matrix 的第 5 行)，这是 [1,5,7]，然后我需要查看 [1,5,7] 中的哪些元素与子矩阵匹配(相当于获得第 5 行的第 1、5 和 7 列)。在这种情况下，第 5 行的结果应该是 [0,1,0,1,0]，因为两个数组中只有 1 和 5 相交)。

如何在给定 adj 的情况下有效地选择此子矩阵。名单？

Answer 1

adj_list = [[2, 6], [1, 3, 24], [2, 4], [3, 5, 21], [4, 6, 10], [1, 5, 7], [6, 8, 9], [7], [7, 10, 14], [5, 9, 11], [10, 12, 18], [11, 13], [12, 14, 15], [9, 13], [13, 16, 17], [15], [15], [11, 19, 20], [18], [18], [4, 22, 23], [21], [21], [2, 25, 26], [24], [24]]

submatrix = (0, 1, 2, 5, 22)

result = [[i in adj_list[sm] for i in submatrix] for sm in submatrix]

应该这样做；尽管我怀疑如果您更仔细地考虑您的最终目标，您可能更愿意计算除此之外的其他东西。