我想计算n
矩阵的下一个相邻元素的乘积。数n
元素相乘应该在函数的输入给出。例如,对于此输入,我应该从第一个开始计算每3个连续元素的乘积。
[p, ind] = max_product([1 2 2 1 3 1],3);
这给了[1*2*2, 2*2*1, 2*1*3, 1*3*1] = [4,4,6,3]
。
有什么实际的方法吗?现在,我使用以下方法执行此操作:
for ii = 1:(length(v)-2)
p = prod(v(ii:ii+n-1));
end
其中v
是输入向量,n
是要相乘的元素数。
在此示例中,n=3
但可以采用任何正整数值。
根据n
是奇数还是偶数还是length(v)
奇数还是偶数,我有时会得到正确的答案,但有时会出错。
例如参数:
v = [1.35912281237829 -0.958120385352704 -0.553335935098461 1.44601450110386 1.43760259196739 0.0266423803393867 0.417039432979809 1.14033971399183 -0.418125096873537 -1.99362640306847 -0.589833539347417 -0.218969651537063 1.49863539349242 0.338844452879616 1.34169199365703 0.181185490389383 0.102817336496793 0.104835620599133 -2.70026800170358 1.46129128974515 0.64413523430416 0.921962619821458 0.568712984110933]
n = 7
我得到了错误:
Index exceeds matrix dimensions.
Error in max_product (line 6)
p = prod(v(ii:ii+n-1));
有什么正确的一般方法吗?
这个方便的解决方案受到了Dev-iL精心思考的答案的启发,不需要Matlab R2016a或更高版本:
out = real( exp(conv(log(a),ones(1,n),'valid')) )
基本思想是将乘法转换为总和,并且可以使用移动平均值,而移动平均值又可以通过conv
合并来实现。
这是gallery
在与元素相乘之前获取循环矩阵并索引所得矩阵的相关部分的一种方法:
a = [1 2 2 1 3 1]
n = 3
%// circulant matrix
tmp = gallery('circul', a(:))
%// product of relevant parts of matrix
out = prod(tmp(end-n+1:-1:1, end-n+1:end), 2)
out =
4
4
6
3
在输入中没有零的情况下,内存效率更高的替代方案:
a = [10 9 8 7 6 5 4 3 2 1]
n = 2
%// cumulative product
x = [1 cumprod(a)]
%// shifted by n and divided by itself
y = circshift( x,[0 -n] )./x
%// remove last elements
out = y(1:end-n)
out =
90 72 56 42 30 20 12 6 2
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句