import numpy as np
a = np.array([3, 1, 2, 4, 6, 1])
b=np.argmax(a)#取出a中元素最大值所對(duì)應(yīng)的索引，此時(shí)最大值位6，其對(duì)應(yīng)的位置索引值為4，（索引值默認(rèn)從0開(kāi)始）
print(b)#4

import numpy as np
a = np.array([[1, 5, 5, 2],
? ? ? ? ? ? ? [9, 6, 2, 8],
? ? ? ? ? ? ? [3, 7, 9, 1]])
b=np.argmax(a, axis=0)#對(duì)二維矩陣來(lái)講a[0][1]會(huì)有兩個(gè)索引方向，第一個(gè)方向?yàn)閍[0]，默認(rèn)按列方向搜索最大值
#a的第一列為1，9，3,最大值為9，所在位置為1，
#a的第一列為5，6，7,最大值為7，所在位置為2，
#此此類(lèi)推，因?yàn)閍有4列，所以得到的b為1行4列，
print(b)#[1 2 2 1]
?
c=np.argmax(a, axis=1)#現(xiàn)在按照a[0][1]中的a[1]方向，即行方向搜索最大值，
#a的第一行為1，5，5，2,最大值為5（雖然有2個(gè)5，但取第一個(gè)5所在的位置），索引值為1，
#a的第2行為9，6，2，8,最大值為9，索引值為0，
#因?yàn)閍有3行，所以得到的c有3個(gè)值，即為1行3列
print(c)#[1 0 2]

import numpy as np
a = np.array([
? ? ? ? ? ? ? [
? ? ? ? ? ? ? ? ? [1, 5, 5, 2],
? ? ? ? ? ? ? ? ? [9, -6, 2, 8],
? ? ? ? ? ? ? ? ? [-3, 7, -9, 1]
? ? ? ? ? ? ? ],
?
? ? ? ? ? ? ? [
? ? ? ? ? ? ? ? ? [-1, 7, -5, 2],
? ? ? ? ? ? ? ? ? [9, 6, 2, 8],
? ? ? ? ? ? ? ? ? [3, 7, 9, 1]
? ? ? ? ? ? ? ],
? ? ? ? ? ? [
? ? ? ? ? ? ? ? ? [21, 6, -5, 2],
? ? ? ? ? ? ? ? ? [9, 36, 2, 8],
? ? ? ? ? ? ? ? ? [3, 7, 79, 1]
? ? ? ? ? ? ? ]
? ? ? ? ? ? ])
b=np.argmax(a, axis=0)#對(duì)于三維度矩陣，a有三個(gè)方向a[0][1][2]
#當(dāng)axis=0時(shí)，是在a[0]方向上找最大值，即兩個(gè)矩陣做比較，具體
#(1)比較3個(gè)矩陣的第一行，即拿[1, 5, 5, 2],
# ? ? ? ? ? ? ? ? ? ? ? ? [-1, 7, -5, 2],
# ? ? ? ? ? ? ? ? ? ? ? ? [21, 6, -5, 2],
#再比較每一列的最大值在那個(gè)矩陣中，可以看出第一列1，-2，21最大值為21，在第三個(gè)矩陣中，索引值為2
#第2列5，7，6最大值為7，在第2個(gè)矩陣中，索引值為1.....，最終得出比較結(jié)果[2 1 0 0]
#再拿出三個(gè)矩陣的第二行，按照上述方法，得出比較結(jié)果 [0 2 0 0]
#一共有三個(gè)，所以最終得到的結(jié)果b就為3行4列矩陣
print(b)
#[[0 0 0 0]
?#[0 1 0 0]
?#[1 0 1 0]]
?
c=np.argmax(a, axis=1)#對(duì)于三維度矩陣，a有三個(gè)方向a[0][1][2]
#當(dāng)axis=1時(shí)，是在a[1]方向上找最大值，即在列方向比較，此時(shí)就是指在每個(gè)矩陣內(nèi)部的列方向上進(jìn)行比較
#(1)看第一個(gè)矩陣
? ? ? ? ? ? ? ? ? # [1, 5, 5, 2],
? ? ? ? ? ? ? ? ? # [9, -6, 2, 8],
? ? ? ? ? ? ? ? ? # [-3, 7, -9, 1]
#比較每一列的最大值，可以看出第一列1，9，-3最大值為9，，索引值為1
#第2列5，-6，7最大值為7，，索引值為2
# 因此對(duì)第一個(gè)矩陣，找出索引結(jié)果為[1，2，0,1]
#再拿出2個(gè)，按照上述方法，得出比較結(jié)果 [1 0 2 1]
#一共有三個(gè)，所以最終得到的結(jié)果b就為3行4列矩陣
print(c)
#[[1 2 0 1]
?# [1 0 2 1]
?# [0 1 2 1]]
?
d=np.argmax(a, axis=2)#對(duì)于三維度矩陣，a有三個(gè)方向a[0][1][2]
#當(dāng)axis=2時(shí)，是在a[2]方向上找最大值，即在行方向比較，此時(shí)就是指在每個(gè)矩陣內(nèi)部的行方向上進(jìn)行比較
#(1)看第一個(gè)矩陣
? ? ? ? ? ? ? ? ? # [1, 5, 5, 2],
? ? ? ? ? ? ? ? ? # [9, -6, 2, 8],
? ? ? ? ? ? ? ? ? # [-3, 7, -9, 1]
#尋找第一行的最大值，可以看出第一行[1, 5, 5, 2]最大值為5，，索引值為1
#第2行[9, -6, 2, 8],最大值為9，，索引值為0
# 因此對(duì)第一個(gè)矩陣，找出行最大索引結(jié)果為[1，0，1]
#再拿出2個(gè)矩陣，按照上述方法，得出比較結(jié)果 [1 0 2 1]
#一共有三個(gè)，所以最終得到的結(jié)果d就為3行3列矩陣
print(d)
# [[1 0 1]
# ?[1 0 2]
# ?[0 1 2]]
###################################################################
#最后一種情況，指定矩陣a[0, -1, :]，第一個(gè)數(shù)字0代表取出第一個(gè)矩陣（從前面可以看出a有3個(gè)矩陣）為
# [1, 5, 5, 2],
# [9, -6, 2, 8],
# [-3, 7, -9, 1]
#第二個(gè)數(shù)字“-1”代表拿出倒數(shù)第一行，為
# [-3, 7, -9, 1]
#這一行的最大索引值為1
?
# ，-1，代表最后一行
m=np.argmax(a[0, -1, :])
print(m)#1
?
#h,取a的第2個(gè)矩陣
# [-1, 7, -5, 2],
# [9, 6, 2, 8],
# [3, 7, 9, 1]
#的第3行
# [3, 7, 9, 1]
#的最大值為9，索引為2
h=np.argmax(a[1, 2, :])
print(h)#2
?
g=np.argmax(a[1,:, 2])#g,取出矩陣a，第2個(gè)矩陣的第3列為-5，2，9,最大值為9，索引為2
print(g)#2