benchmarks.wiki

benchmarks.wiki / Public workspace

DS-1000 / 437 / Input example:

Problem

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

prompt

Problem:
Input example:
I have a numpy array, e.g.
a=np.array([[0,1], [2, 1], [4, 8]])
Desired output:
I would like to produce a mask array with the min value along a given axis, in my case axis 1, being True and all others being False. e.g. in this case
mask = np.array([[True, False], [False, True], [True, False]])
How can I achieve that?

A:
<code>
import numpy as np
a = np.array([[0, 1], [2, 1], [4, 8]])
</code>
mask = ... # put solution in this variable
BEGIN SOLUTION
<code>

library

Numpy

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
437
task_id
6b78c81b-3d19-5567-8958-d3cf819ec889
task_key
default--test--437
task_revision_id
1
config / split
default / test

Official source

initial import