numpy.ndarray数组求和axis=n的问题
发布时间
阅读量：
阅读量
关于numpy.ndarray数组在执行sum(axis=n)操作时所涉及的轴向处理问题
实验
import numpy as np
arr = np.array([[[0, 1, 2, 3],
[0, 1, 2, 3],
[0, 1, 2, 3]],
[[0, 1, 2, 3],
[0, 1, 2, 3],
[0, 1, 2, 3]]])
推测
arr是一个形状为(2,3,4)的数组，
当axis参数设置为0时，运算后的输出数组应具有(3,4)的维度
当axis参数设置为1时，运算后的输出数组应具有(2,4)的维度
当axis参数设置为2时，运算后的输出数组应具有(2,3)的维度
验证
对数组沿第0轴进行求和运算
所得结果为(3,4):
array([[0, 2, 4, 6],
[0, 2, 4, 6],
[0, 2, 4, 6]全部评论 (0)
还没有任何评论哟~