版权声明: https://blog.csdn.net/dashoumeixi/article/details/86568375
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))