問題描述： 給定一個二維陣列，求每一行的最大值 返回一個列向量

如： 給定陣列【1,2，3；4,5，3】 返回[3；5]

import numpy as np

x = np.array([[1,2,3],[4,5,3]])
# 先求每行最大值得下標
index_max = np.argmax(x, axis=1)# 其中，axis=1表示按行計算
print(index_max.shape)

max = x[range(x.shape[0]), index_max]
print(max)
# 注意到這裡返回的是行向量
# 這可以是一種通用的方法，
# 其中range（）可以是一個列向量，表示0到n
# index_max也是一個列向量，表示具體的座標
# 這樣，兩個座標組合起來就成為了二維索引

max_ = x[range(x.shape[0]), index_max].reshape(-1,1)
print(max_)
# 這樣變成了列向量
 

值得注意的是： 1）np.argmax得到的是列向量，而不是行向量，這在其他的函式中也有體現 2）求和以及其他運算可以按照行或者列來，通過指定axis即可 3）通過reshape()來重新返回具體的維度，我們需要的維度。函式的引數可以有一個-1，但只能有一個，表示這個數是未知的