np.meshgrid 根据子矩阵信息构建矩阵
面临类似以子矩阵替换原始矩阵元素的矩阵构建方法在numpy以及r中都很少有方便的实现,以下提供一个较为快捷的方法。
问题简述:
全程用for循环可能因为python程序执行原因导致时间开销大,尝试通过numpy内置c语言方法加快构建速度。具体实现方法如下:
a0 = np.array([1,2,2,1])
idx_list = [idx+1 for idx,value in enumerate(a0) for i in range(value)]
nd0 = np.meshgrid(idx_list,idx_list,sparse=True)
a1 = nd0[0] / nd0[1]
a1[a1 != 1] = 0