python - 加速Python中过滤矩阵的代码

Question

我有一个来自 [[row1],[row2] 等] 的对称矩阵 d1，我想从中删除一些元素。我要删除的元素的索引在列表 ind 中指示(例如 ind = [1,2,3] 表示我要删除 d1[1][1],d1[2][2] 和 d1[3 ][3])。

我编写了以下代码，将过滤后的矩阵 d1 写入 d2 中。然而，它需要永远运行。有什么方法可以加快该代码的速度吗？

 for k in range(len(d1)):
     if k not in ind:
         d2.append([])
         for j in range(len(d1)):
             if j not in ind:
                 d2[-1].append(d1[k][j])

 print(d2)
 return d2

示例:

d1 = [[1, 2, 3,6,8],[4,5,6,6,6],[7,8,9,6,6],[1, 2, 3,6,6],[1,2,3,6,9]]

ind = [0,3]

d2 = [[5,6,6],[8,9,6],[2, 3,9]]

Answer 1

主要问题可能是，如果 ind 是一个非常长的列表(数千个条目)，那么每次执行 if x not in ind 时，您都必须检查< em>整个列表。只需将 ind 更改为 set 可能会大大加快速度。此外，您可以创建一个要保留的良好索引列表，而不是检查两个循环中的条件。最后，使其成为列表理解可能会加快速度并使其更具可读性。

ind_set = set(ind)
retain = [i for i in range(len(d1)) if i not in ind_set]
d2 = [[d1[k][j] for j in retain] for k in retain]