举个含有axis参数的例子
时间: 2023-09-18 20:09:22 浏览: 42
好的,下面是一个含有axis参数的例子:
```python
import numpy as np
a = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
cumulative_sum = np.cumsum(a, axis=0)
print(cumulative_sum)
```
输出结果为:
```
[[ 1 2 3]
[ 5 7 9]
[12 15 18]]
```
在这个例子中,我们创建了一个3x3的二维数组a,然后指定axis=0参数,表示按列计算累积和。结果数组cumulative_sum的第一行和a的第一行相同,第二行是第一行和第二行的和,第三行是前面两行和第三行的和。