避免使用numpy矩阵进行for循环

莫瑟

我想知道这一定是在没有for循环的情况下执行以下操作的方法

import numpy
from itertools import product as itprod

a = np.arange(120.).reshape(3,2,5,2,2)
fact = np.linspace(1,1.4,15).reshape((3,5))

for i,j in itprod(range(3),range(5)):
    a[i,:,j]*= fact[i,j]

有什么建议??

斯文·马纳赫(Sven Marnach)

要利用广播的优势,您必须fact在正确的位置插入新的坐标轴

a *= fact[:, np.newaxis, :, np.newaxis, np.newaxis] 

本文收集自互联网,转载请注明来源。

如有侵权,请联系[email protected] 删除。

编辑于
0

我来说两句

0条评论
登录后参与评论

相关文章

来自分类Dev

避免使用 numpy 向量的 for 循环

来自分类Dev

通过避免使用for循环来定义矩阵

来自分类Dev

使用numpy避免循环或列表理解

来自分类Dev

优化numpy矩阵运算(当前使用for循环)

来自分类Dev

优化numpy矩阵运算(当前使用for循环)

来自分类Dev

使用循环和条件进行矩阵计算

来自分类Dev

避免在R中使用多个for循环来计算矩阵

来自分类Dev

如何避免对每个元素使用函数来循环变换矩阵

来自分类Dev

如何使用循环对矩阵内的矩阵进行编码?

来自分类Dev

使用numpy数组进行矩阵乘法

来自分类Dev

使用numpy进行类似的矩阵计算

来自分类Dev

删除numpy中的循环以进行简单的矩阵分配

来自分类Dev

numpy按循环逐列对矩阵进行排序

来自分类Dev

如何避免稀疏矩阵的循环?

来自分类Dev

如何避免稀疏矩阵的循环?

来自分类Dev

是否可以使用numpy避免此循环?

来自分类Dev

使用 numpy 避免在 2D 数组上循环

来自分类Dev

尝试避免使用sapply进行for循环(对于gsub)

来自分类Dev

使用OpenMP(C)进行矩阵乘法-折叠所有循环

来自分类Dev

在列中使用矩阵累加值进行循环

来自分类Dev

NumPy:如何避免这种循环?

来自分类Dev

避免对类进行循环引用

来自分类Dev

在矩阵索引求和时避免嵌套循环

来自分类Dev

使用 Numpy 进行矩阵运算的更简单方法

来自分类Dev

循环遍历 numpy 矩阵元素

来自分类Dev

避免使用for循环

来自分类Dev

在jQuery中使用回调函数进行循环或如何避免循环

来自分类Dev

使用来自熊猫的数据进行numpy循环

来自分类Dev

使用numpy更新节点值而不进行for循环