python 获取数组维度

采用.ndim来获取数组维度
x = np.array([1, 2])
print(x.ndim)
return 1

x = np.array([[1, 2], [1,2]])
print(x.ndim)
return 2

argmax() 给出第一个最大值的索引，数组可转成bool查找第一个满足条件的索引
x = np.array([1, 2, 3, 5])
argmax(x > 2)
return 2