在python中创建均匀间隔的网格

时间:2018-04-11 03:32:33

标签: python numpy

我是python的新手,我知道必须有更好的方法来做到这一点,尤其是numpy,并且不会附加到数组。有没有更简洁的方法在python中做这样的事情?

def create_uniform_grid(low, high, bins=(10, 10)):
    """Define a uniformly-spaced grid that can be used to discretize a space.

    Parameters
    ----------
    low : array_like
        Lower bounds for each dimension of the continuous space.
    high : array_like
        Upper bounds for each dimension of the continuous space.
    bins : tuple
        Number of bins along each corresponding dimension.

    Returns
    -------
    grid : list of array_like
        A list of arrays containing split points for each dimension.
    """
    range1 = high[0] - low[0]
    range2 = high[1] - low[1]

    steps1 = range1 / bins[0]
    steps2 = range2 / bins[1]

    arr1 = []
    arr2 = []

    for i in range(0, bins[0] - 1):
        if(i == 0):
            arr1.append(low[0] + steps1)
            arr2.append(low[1] + steps2)
        else:
            arr1.append(round((arr1[i - 1] + steps1), 1))
            arr2.append(arr2[i - 1] + steps2)

    return [arr1, arr2]


low = [-1.0, -5.0]
high = [1.0, 5.0]
create_uniform_grid(low, high)

# [[-0.8, -0.6, -0.4, -0.2, 0.0, 0.2, 0.4, 0.6, 0.8],
# [-4.0, -3.0, -2.0, -1.0, 0.0, 1.0, 2.0, 3.0, 4.0]]

2 个答案:

答案 0 :(得分:1)

np.ogrid与您的功能类似。差异:1)它将保留端点; 2)它将创建一个列和一行,因此它的输出是'广播就绪':

    >>> np.ogrid[-1:1:11j, -5:5:11j]
[array([[-1. ],
       [-0.8],
       [-0.6],
       [-0.4],
       [-0.2],
       [ 0. ],
       [ 0.2],
       [ 0.4],
       [ 0.6],
       [ 0.8],
       [ 1. ]]), array([[-5., -4., -3., -2., -1.,  0.,  1.,  2.,  3.,  4.,  5.]])]

答案 1 :(得分:0)

也许getData() { this._dataService.data.subscribe((result) => { this.data = result; console.log(this.data); }); } 就是你想要的。

以下是创建网格并对其进行数学运算的示例:

numpy.meshgrid

enter image description here

参见:

  1. https://docs.scipy.org/doc/numpy/reference/generated/numpy.meshgrid.html