benchmarks.wiki

benchmarks.wiki / Public workspace

DS-1000 / 307 / Following-up from this question years ago, is there a "shift" function in numpy?…

Problem

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

prompt

Problem:
Following-up from this question years ago, is there a "shift" function in numpy? Ideally it can be applied to 2-dimensional arrays, and the numbers of shift are different among rows.
Example:
In [76]: xs
Out[76]: array([[ 0.,  1.,  2.,  3.,  4.,  5.,  6.,  7.,  8.,  9.],
		 [ 1.,  2.,  3.,  4.,  5.,  6.,  7.,  8.,  9., 10.]])
In [77]: shift(xs, [1,3])
Out[77]: array([[nan,   0.,   1.,   2.,   3.,   4.,   5.,   6.,	7.,	8.], [nan, nan, nan, 1.,  2.,  3.,  4.,  5.,  6.,  7.])
In [78]: shift(xs, [-2,-3])
Out[78]: array([[2.,   3.,   4.,   5.,   6.,   7.,   8.,   9.,  nan,  nan], [4.,  5.,  6.,  7.,  8.,  9., 10., nan, nan, nan]])
Any help would be appreciated.
A:
<code>
import numpy as np
a = np.array([[ 0.,  1.,  2.,  3.,  4.,  5.,  6.,  7.,  8.,  9.],
		[1.,  2.,  3.,  4.,  5.,  6.,  7.,  8.,  9., 10.]])
shift = [-2, 3]
</code>
result = ... # 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
307
task_id
baf31826-98a8-5f0b-8b3a-b2bf95fff691
task_key
default--test--307
task_revision_id
1
config / split
default / test

Official source

initial import