请问sma = np.convolve(weights,end_price)[N-1:-N+1]中的[N-1:-N+1]如何理解

Question

请问sma = np.convolve(weights,end_price)[N-1:-N+1]中的[N-1:-N+1]如何理解

Answer 1

meteor2022 2024-03-18 07:56:20

[N-1:-N+1]这块实际上和卷积的计算原理有关。

weights 为1*N矩阵，end_price为1*M矩阵，M > N

案例中我们实现的是full-mode卷积，该模式的全部结果返回N+M-1维的数组。

根据卷积的运算结果只有在[N-1:-N+1]时，才会weights中所有的元素都被计算到，也就是N日的EMA。

两个一维矩阵的卷积计算原理，这篇资料说的很详细，可以参考https://blog.csdn.net/weixin_41584472/article/details/126134540