我想执行相当于 numpy
A[R][:,R]
但在纯 python 中。例如:
A = [[0, 1, 2], [1, 2, 3], [2, 3, 4]] 和
R = [1, 2, 2]
输出应该是:
[[2, 3, 3],[3, 4, 4], [3, 4, 4]]
在纯 python 中有没有好的方法来做到这一点？
请您参考如下方法:
我猜那将是[[A[i][k] for k in R] for i in R]
>>> A = [[0, 1, 2], [1, 2, 3], [2, 3, 4]] ; R = [1, 2, 2]
>>> np.array(A)[R][:,R]
array([[2, 3, 3],
[3, 4, 4],
[3, 4, 4]])
>>> [[A[i][k] for k in R] for i in R]
[[2, 3, 3], [3, 4, 4], [3, 4, 4]]
