美文网首页
numpy sum axis解释

numpy sum axis解释

作者: help_youself | 来源:发表于2021-02-23 15:29 被阅读0次

 关于a.sum(axis=1),网上有很多博客解释,但都说的不明白。我在stackoverflow上发现一个解释[1],很容易记忆,录在这里。
 code.py

import  numpy as np
if __name__ == '__main__':
    a = np.arange(30).reshape(2, 3, 5)
    print(a)
    print(a[0,:,:]+a[1,:,:])
    print("\n")
    print(a.sum(axis=0))
    print("\n\n")
    print(a[:,0,:]+a[:,1,:]+a[:,2,:])
    print("\n")
    print(a.sum(axis=1))
    
    print("\n\n")
    print(a[:,:,0]+a[:,:,1]+a[:,:,2]+a[:,:,3]+a[:,:,4])
    print("\n")
    print(a.sum(axis=2))

 a.sum(axis=0)的结果同a[0,:,:]+a[1,:,:]相同。
 a.sum(axis=1)的结果同a[:,0,:]+a[:,1,:]+a[:,2,:]相同。
 a.sum(axis=2)的结果同a[:,:,0]+a[:,:,1]+a[:,:,2]+a[:,:,3]+a[:,:,4]相同。

Reference:
[1] How does NumPy Sum (with axis) work?

相关文章

网友评论

      本文标题:numpy sum axis解释

      本文链接:https://www.haomeiwen.com/subject/mrgrfltx.html