numpy argmax 随机漫步

argmax 获取最大的索引值;

data = np.array([1,3,1,3,2,4,2])

print(data.argmax(0))  //获取最大索引

#先获取一个只有0,1的数组, 大小(5,4) 可自行调整
arr = np.random.randint(0,2,(5,4))

#把 > 0 的用1代替, 否则用-1 代替  :  1 if arr > 0 else -1
arr = np.where(arr > 0 , 1, -1)
print("arr:",arr)

print("sum:" , data.sum())

#获取累加过程
data = arr.cumsum(1)
print("cumsum:",data)

# 提取 > 0的子数组
hint1 = (data > 0).any(1)
print(">0 sub arr:", hint1)

print(data[hint1].argmax(1))