如何在Python中使用numpy获得累计最大索引?

我在试着 x 和 y 向量中所有累积最大值的坐标。我写了一篇天真的 for-loop 但我觉得有办法 numpy 那更优雅。

有人知道 努比 能做到这一点吗?

细节应通过以下曲线图进行描述:

# Generate data
x = np.abs(np.random.RandomState(0).normal(size=(100)))
y = np.linspace(0,1,x.size)
z = x*y

# Get maximiums
idx_max = list()
val_max = list()

current = 0
for i,v in enumerate(z):
    if v > current:
        idx_max.append(i)
        val_max.append(v)
        current = v

# Plot them
with plt.style.context("seaborn-white"):
    fig, ax = plt.subplots(figsize=(10,3), ncols=2)
    ax[0].plot(z)
    ax[1].plot(z, alpha=0.618)
    ax[1].scatter(idx_max, val_max, color="black", edgecolor="maroon", linewidth=1.618)