将numpy的cumprod功能与矩阵乘法结合使用

2024-05-14 23:11:27 发布

您现在位置:Python中文网/ 问答频道 /正文

我正在尝试使用numpy的cumprod的功能,但是用于矩阵。 有一个充满齐次变换的矩阵(为了简单起见,这里是随机的),我想使用numpy的矩阵乘法来累积变换

n = 1000
data = np.random([n, 4,4])
dataAcc = cumProd(data, np.matmul)

dataAcc应包含以下内容:

dataAcc[0,:,:] = data[0,:,:]
dataAcc[1,:,:] = data[0,:,:] @ data[1,:,:] = dataAcc[0,:,:] @ data[1,:,:]
dataAcc[2,:,:] = dataAcc[1,:,:] @ data[2,:,:]
...

有没有一种方法可以使用快速numpy函数或其他等效函数来实现这一点? 我不想使用循环

干杯


Tags: 方法函数功能numpydatanp矩阵random

热门问题