Using numpy mgrid with a variable number of indices

Question

How do you use numpy.mgrid with a variable number of indices? I can't find any examples on github of anyone using this with anything but hardcoded values.

import numpy as np
np.mgrid[1:10, 1:10] # this works fine

x = (1, 10)
np.mgrid[x[0]:x[1], x[0]:x[1]] # hardcoded

xs = [(1,10)] * 10
np.mgrid[*xs????] # I can't get anything to work here

heltonbiker · Accepted Answer · 2015-02-24 00:52:59Z

4

This seems to work:

np.mgrid[[slice(i,j) for i,j in [(1,10)]*10]]

though with *10 it is too large

It's derived from that fact

np.mgrid[slice(1,10),slice(1,10)]  # same as
np.mgrid[1:10,1:10]

edited Feb 24, 2015 at 0:52

heltonbiker

27.7k31 gold badges151 silver badges270 bronze badges

answered Feb 24, 2015 at 0:35

hpaulj

233k14 gold badges260 silver badges392 bronze badges

Sign up to request clarification or add additional context in comments.

3 Comments

heltonbiker Over a year ago

Would it be correct to say that the notation in your last line is just "syntactic sugar" to create the slice objects?

Charles Offenbacher Over a year ago

Great, thanks! Never had the occasion to use slice() before, nice.

hpaulj Over a year ago

Yes, the Python interpreter converts the [x:y:z] to a slice object. docs.python.org/2/reference/…

Collectives™ on Stack Overflow

Using numpy mgrid with a variable number of indices

1 Answer 1

3 Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

3 Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related