如何理解数组的行数和列数,差点搞蒙?

最近接触数组,就这个行和列有时候傻傻分不清楚,因为维度太高。今天看一个简单的小例子,即可搞懂咋回事!

import numpy as np
np.random.seed(10)
a=np.random.randint(1,10,[5,3])
print(a,np.amax(a,axis=0),a.argmax(axis=0))  
print(a,np.amax(a,axis=1),a.argmax(axis=1))

输出:

a=[[5 1 2]
 [1 2 9]
 [1 9 7]
 [5 4 1]
 [5 7 9]]

a是一个5行3列的二维数组。

np.amax(a,axis=0)=[5 9 9]   #axis=0,表示取出每列的最大值
a.argmax(axis=0))=[0 2 1]   #每列最大值的索引
np.amax(a,axis=1)=[5 9 9 5 9]  #axis=1,表示取出每行的最大值
a.argmax(axis=1)= [0 2 1 0 2]   #每行最大值的索引
print(a.sum(axis=1))  #[ 8 12 17 10 21]  axis=1表示对每行求和
print(a.sum(axis=0))  #[17 23 28]        axis=0表示对每列求和
print(np.size(a,0))   # 5 返回其行数
print(np.size(a,1))   # 3 返回其列数

看了这个例子,应该清楚了吧。

import numpy as np
np.random.seed(10)
a=np.random.randint(1,10,[5,3])
print(a,np.amax(a,axis=0),a.argmax(axis=0))
print(a,np.amax(a,axis=1),a.argmax(axis=1))
print(a.sum(axis=1))
print(a.sum(axis=0))

至于输出结果自己运行下,就理解了。

Original: https://blog.csdn.net/weixin_48266700/article/details/123669060
Author: XUE FENG!
Title: 如何理解数组的行数和列数,差点搞蒙?

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/761435/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球