Convert coordinates in a region into a numpy array in Python

Question

I have a rectangular region, where the lower left (min) and upper right (max) coordinates. I want to make a Numpy array that consists of coordinates in the region. For example, the min and max are (3, 8) and max (0, 6), respectively. The numpy array I want to make is

Additionally, I have a requirement that the grid in the region may be smaller than 1. For example, 0.5 makes X = [[3.0, 0] [3.5, 0] [4.0, 0] [4.5, 0] ...]]

kennytm · Accepted Answer · 2017-03-30 09:18:20Z

4

You could use numpy.mgrid:

>>> numpy.mgrid[3:8, 0:6].T
array([[[3, 0],
        [4, 0],
        [5, 0],
        [6, 0],
        [7, 0]],

       [[3, 1],
        [4, 1],
        [5, 1],
        [6, 1],
        [7, 1]],

        ...

       [[3, 5],
        [4, 5],
        [5, 5],
        [6, 5],
        [7, 5]]])

If you want a "flat" array of tuples, you can reshape it:

>>> numpy.mgrid[3:8, 0:6].T.reshape((-1, 2))
array([[3, 0],
       [4, 0],
       [5, 0],
       [6, 0],
       [7, 0],
       [3, 1],
       ...
       [6, 5],
       [7, 5]])

answered Mar 30, 2017 at 9:18

kennytm

526k110 gold badges1.1k silver badges1k bronze badges

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

2 Comments

Youngsup Kim Over a year ago

Wow, it is wonderful. Is there a way to set a step size to be smaller than 1?

kennytm Over a year ago

@YoungsupKim Yes. numpy.mgrid[0.0:1.0:0.1, 3.5:4.5:0.2].T.reshape((-1, 2)).

Collectives™ on Stack Overflow

Convert coordinates in a region into a numpy array in Python

1 Answer 1

2 Comments

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

2 Comments

Related