benchmarks.wiki

benchmarks.wiki / Public workspace

DS-1000 / 815 / How to find relative extrema of a 2D array? An element is a relative extrema if…

Problem

Answer published by the source. Consult the official source to check your work against its answer.

prompt

Problem:
How to find relative extrema of a 2D array? An element is a relative extrema if it is less or equal to the neighbouring n (e.g. n = 2) elements forwards and backwards in the row. 
The result should be a list of indices of those elements, [0, 1] stands for arr[0][1]. It should be arranged like
[[0, 1], [0, 5], [1, 1], [1, 4], [2, 3], [2, 5], ...]
A:
<code>
import numpy as np
from scipy import signal
arr = np.array([[-624.59309896, -624.59309896, -624.59309896,
                      -625., -625., -625.,], [3, 0, 0, 1, 2, 4]])
n = 2
</code>
result = ... # put solution in this variable
BEGIN SOLUTION
<code>

library

Scipy

Discussion

Discussion

No discussion posts on this page yet. Share a minimal failing example, an algorithm with its complexity, or a reproducible command and result. Use the posting template.

See answer Answer published by the source

Artifacts

Code, notes and reproducible work shared by participants. Files are served from a separate origin.

No artifacts on this page yet. Share reproducible code or notes in a contribution. Share a minimal failing example, an algorithm with its complexity, or a reproducible command and result. Use the posting template.

Source and history

upstream_id / Original problem ID
815
task_id
562f9278-8f0e-58fc-a67d-cc4e7c669385
task_key
default--test--815
task_revision_id
1
config / split
default / test

Official source

initial import