benchmarks.wiki

benchmarks.wiki / Public workspace

DS-1000 / 449 / I have two 2D numpy arrays like this, representing the x/y distances between…

Problem

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

prompt

Problem:
I have two 2D numpy arrays like this, representing the x/y distances between three points. I need the x/y distances as tuples in a single array.
So from:
x_dists = array([[ 0, -1, -2],
                 [ 1,  0, -1],
                 [ 2,  1,  0]])
y_dists = array([[ 0, -1, -2],
                 [ 1,  0, -1],
                 [ 2,  1,  0]])
I need:
dists = array([[[ 0,  0], [-1, -1], [-2, -2]],
               [[ 1,  1], [ 0,  0], [-1, -1]],
               [[ 2,  2], [ 1,  1], [ 0,  0]]])
I've tried using various permutations of dstack/hstack/vstack/concatenate, but none of them seem to do what I want. The actual arrays in code are liable to be gigantic, so iterating over the elements in python and doing the rearrangement "manually" isn't an option speed-wise.
A:
<code>
import numpy as np
x_dists = np.array([[ 0, -1, -2],
                 [ 1,  0, -1],
                 [ 2,  1,  0]])

y_dists = np.array([[ 0, -1, -2],
                 [ 1,  0, -1],
                 [ 2,  1,  0]])
</code>
dists = ... # 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
449
task_id
776174ed-7464-5cb8-9c5d-51c95b7acf17
task_key
default--test--449
task_revision_id
1
config / split
default / test

Official source

initial import