np.meshgrid 根据子矩阵信息构建矩阵

idiotic_bird

已于 2022-04-11 16:29:16 修改

阅读量2k

点赞数

分类专栏： python 问题集文章标签： python

于 2022-04-11 16:28:30 首次发布

本文链接：https://blog.csdn.net/idiotic_bird/article/details/124102215

版权

python 问题集专栏收录该内容

4 篇文章 0 订阅

订阅专栏

这篇博客介绍了如何使用numpy的meshgrid函数来快速构建矩阵，特别是在需要根据子矩阵信息替换原始矩阵元素的场景下。通过使用for循环可能导致效率低下，因此作者提出了一种利用numpy的内置C语言加速的方法。文章中展示了如何通过enumerate和range函数生成索引列表，然后利用meshgrid创建网格，并进行除法操作得到新矩阵a1，最后将非1元素置为0，从而实现特定的矩阵构建需求。

摘要由CSDN通过智能技术生成

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