Julia累加和累乘

文章目录

累加和累乘

cumsumcumprod分别用于累加或者累乘,其效果如下

python 复制代码
x = ones(Int, 1,5)      # 1  1  1  1  1
y = cumsum(x, dims=2)   # 1  2  3  4  5
z = cumprod(y, dims=2)  # 1  2  6  24  120

accumulate

accumulate可基于任意运算符对数组进行计算,例如

python 复制代码
accumulate(+, x)    # 1  2  3  4  5

如果计算的是矩阵,那么在未声明轴的情况下,默认按列累积

js 复制代码
x = ones(Int, 5, 5);
accumulate(+, x)

上述代码的计算结果如下

[ 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ] → [ 1 6 11 16 21 2 7 12 17 22 3 8 13 18 23 4 9 14 19 24 5 10 15 20 25 ] \begin{bmatrix} 1&1&1&1&1\\1&1&1&1&1\\ 1&1&1&1&1\\1&1&1&1&1\\ 1&1&1&1&1\\ \end{bmatrix}\to \begin{bmatrix} 1 &6&11&16&21\\2& 7&12&17&22\\ 3& 8&13&18&23\\4& 9&14&19&24\\ 5&10&15&20&25\\ \end{bmatrix} 1111111111111111111111111 → 12345678910111213141516171819202122232425

如果指定坐标轴,那么将沿着坐标轴进行计算,例如下面的代码

js 复制代码
accumulate(+, x, dims=1)

计算结果如下

[ 1 1 1 1 1 2 2 2 2 2 3 3 3 3 3 4 4 4 4 4 5 5 5 5 5 ] \begin{bmatrix} 1&1&1&1&1\\2&2&2&2&2\\ 3&3&3&3&3\\4&4&4&4&4\\ 5&5&5&5&5\\ \end{bmatrix} 1234512345123451234512345

accumulate还可以指定初始值,例如

js 复制代码
accumulate(+, x, init=100)

计算结果为

[ 101 106 111 116 121 102 107 112 117 122 103 108 113 118 123 104 109 114 119 124 105 110 115 120 125 ] \begin{bmatrix} 101&106&111&116&121\\102&107&112&117&122\\ 103&108&113&118&123\\104&109&114&119&124\\ 105&110&115&120&125\\ \end{bmatrix} 101102103104105106107108109110111112113114115116117118119120121122123124125

!函数

cumsum, cumprod以及accumulate都有一个配套的!函数,可以直接更改输入变量的值,其参数列表如下,其中参数A是待处理数组,B则用于存放计算结果。

js 复制代码
cumsum!(B, A; dims::Integer)
cumprod!(B, A; dims::Integer)
accumulate!(op, B, A; [dims], [init])
相关推荐
XiaoLeisj34 分钟前
【JavaEE初阶 — 多线程】单例模式 & 指令重排序问题
java·开发语言·java-ee
励志成为嵌入式工程师2 小时前
c语言简单编程练习9
c语言·开发语言·算法·vim
捕鲸叉2 小时前
创建线程时传递参数给线程
开发语言·c++·算法
A charmer2 小时前
【C++】vector 类深度解析:探索动态数组的奥秘
开发语言·c++·算法
Peter_chq2 小时前
【操作系统】基于环形队列的生产消费模型
linux·c语言·开发语言·c++·后端
记录成长java4 小时前
ServletContext,Cookie,HttpSession的使用
java·开发语言·servlet
前端青山4 小时前
Node.js-增强 API 安全性和性能优化
开发语言·前端·javascript·性能优化·前端框架·node.js
睡觉谁叫~~~4 小时前
一文解秘Rust如何与Java互操作
java·开发语言·后端·rust
音徽编程4 小时前
Rust异步运行时框架tokio保姆级教程
开发语言·网络·rust
观音山保我别报错4 小时前
C语言扫雷小游戏
c语言·开发语言·算法