8

我在一个项目中使用 Numpy 和 Python，其中 2D 地图由ndarray：

map = [[1,2,3,4,5],
       [2,3,4,2,3],
       [2,2,2,1,2],
       [3,2,1,2,3],
       [4,6,5,7,4]]
MAP_WIDTH = 5, MAP_HEIGHT = 5

一个对象有一个元组位置：

actor.location = (3,3)

和一个视图范围：

actor.range = 2

如何编写函数actor.view_map(map)，以便地图返回演员位置周围的区域，直至某个范围。例如（使用上面的地图），

range = 1
location = (3, 2)
=>
[[2,3,4],
 [3,4,2],
 [2,2,1]]

但如果演员的范围延伸得太远，我希望地图充满-1：

range = 1
location = (1,1)
[[-1,-1,-1],
 [-1, 1, 2],
 [-1, 2, 3]]

最简单的情况是范围为 0，它返回当前正方形：

range = 0
location = (1, 2)
[[2]]

如何将我的地图切片到某个边界？

4

2 回答 2

4

所以，感谢 Joe Kington，我在我的地图周围添加了一个边框（用 -1 填充）。

这是我的做法，但这可能不是很 Pythonic，因为我刚刚开始使用语言/库：

map = numpy.random.randint(10, size=(2 * World.MAP_WIDTH, 2 * World.MAP_HEIGHT))
map[0 : World.MAP_WIDTH / 4, :] = -1
map[7 * World.MAP_WIDTH / 4 : 2 * World.MAP_WIDTH, :] = -1
map[:, 0 : World.MAP_HEIGHT / 4] = -1
map[:, 7 * World.MAP_HEIGHT / 4 : 2 * World.MAP_WIDTH] = -1
于 2013-04-07T21:28:51.450 回答
2

Box这是一个让使用盒子更容易的小课程——

from __future__ import division
import numpy as np

class Box:
    """ B = Box( 2d numpy array A, radius=2 )
        B.box( j, k ) is a box A[ jk - 2 : jk + 2 ] clipped to the edges of A
        @askewchan, use np.pad (new in numpy 1.7):
            padA = np.pad( A, pad_width, mode="constant", edge=-1 )
            B = Box( padA, radius )
    """

    def __init__( self, A, radius ):
        self.A = np.asanyarray(A)
        self.radius = radius

    def box( self, j, k ):
        """ b.box( j, k ): square around j, k clipped to the edges of A """
        return self.A[ self.box_slice( j, k )]

    def box_slice( self, j, k ):
        """ square, jk-r : jk+r clipped to A.shape """
            # or np.clip ?
        r = self.radius
        return np.s_[ max( j - r, 0 ) : min( j + r + 1, self.A.shape[0] ),
                       max( k - r, 0 ) : min( k + r + 1, self.A.shape[1] )]

#...............................................................................
if __name__ == "__main__":
    A = np.arange(5*7).reshape((5,7))
    print "A:\n", A
    B = Box( A, radius=2 )
    print "B.box( 0, 0 ):\n", B.box( 0, 0 )
    print "B.box( 0, 1 ):\n", B.box( 0, 1 )
    print "B.box( 1, 2 ):\n", B.box( 1, 2 )
于 2013-04-08T11:34:28.627 回答