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))