最大值（amax）变为numpy范围

Question

我有一个numpy数组，我需要使用每个＆n;＆＃39;阵列的位置。

让我试一试：

考虑我有1D numpy数组，其值为1到5.我想要一个新的numpy数组，其中包含原始numpy数组中每3个位置的amax或argmax。

原始numpy数组和amax动作范围：

                  | 1 | 2 | 3 | 4 | 5 |
                  ------|         5 |--
                  3 |-------|
                      4 |-------|
                          5 |-------|
                  --|         5 |-----

我的新numpy数组，考虑到每个3个位置的amax将是（[5,3,4,5,5]）。

任何人都可以想到使用numpy这样做的方法吗？

我能够使用＆＃39; for＃39;但我不认为这是最好的选择。以下是我使用“＆＃39;：

”所做的事情的片段

    for i in range(array-1):
        best = np.argmax([array[i-1], array[i], array[i+1,2]])
        best_array[i] = array[i+best-1]