我正在寻找一种 Numpy(即希望更快)的方式来执行以下操作:
import numpy as np
x = np.array([1,2,3,4,5],dtype=np.double)
arr = [[1,2],[0,4,3],[1,4,0],[0,3,4],[1,4]]
ans = np.array([ x[item] - x[i] for i, item in enumerate(arr) ])
我想摆脱列表理解并做这样的事情(虽然,我知道这行不通)
x[arr[:]] - x[:]
arr
始终是长度等于 的长度的整数的嵌套列表x
。内部列表的长度不一定相同(即arr
是一个参差不齐的列表)