我在Python(2.7)类中有一个函数,它应该在2维numpy数组中检索它周围的'cells'的值。如果索引超出范围,我将该值设置为None。
我正在努力寻找一种方法来做到这一点,而无需编写8个try / catch语句,或者不使用多个if x else None语句,如下面的代码中所示。虽然他们都会工作,但他们似乎没有很好的结构,我认为必须有一个更简单的方法来做到这一点 - 我可能会以完全错误的方式思考这个问题。任何帮助将不胜感激。
# This will return a dictionary with the values of the surrounding points
def get_next_values(self, column, row):
if not (column < self.COLUMNS and row < self.ROWS):
print "Invalid row/column."
return False
nextHorizIsIndex = True if column < self.COLUMNS - 2 else False
nextVertIsIndex = True if row < self.ROWS - 2 else False
n = self.board[column, row-1] if column > 0 else None
ne = self.board[column+1, row-1] if nextHorizIsIndex else None
e = self.board[column+1, row] if nextHorizIsIndex else None
se = self.board[column+1, row+1] if nextHorizIsIndex and nextVertIsIndex else None
s = self.board[column, row+1] if nextVertIsIndex else None
sw = self.board[column-1, row+1] if nextVertIsIndex else None
w = self.board[column-1, row] if row > 0 else None
nw = self.board[column-1, row-1] if 0 not in [row, column] else None
# debug
print n, ne, e, se, s, sw, w, nw
答案 0 :(得分:5)
这是一个标准技巧:创建一个边缘填充值为None的电路板。然后,您可以访问任何内部3x3方格,并使用
填写适当的值nw, n, ne, w, _, e, sw, s, se = (self.board[column-1:column+2, row-1:row+2]).ravel()
例如,
import numpy as np
board = np.empty((10,10), dtype = 'object')
board[:,:] = None
board[1:9, 1:9] = np.arange(64).reshape(8,8)
print(board)
# [[None None None None None None None None None None]
# [None 0 1 2 3 4 5 6 7 None]
# [None 8 9 10 11 12 13 14 15 None]
# [None 16 17 18 19 20 21 22 23 None]
# [None 24 25 26 27 28 29 30 31 None]
# [None 32 33 34 35 36 37 38 39 None]
# [None 40 41 42 43 44 45 46 47 None]
# [None 48 49 50 51 52 53 54 55 None]
# [None 56 57 58 59 60 61 62 63 None]
# [None None None None None None None None None None]]
column = 1
row = 1
nw, n, ne, w, _, e, sw, s, se = (board[column-1:column+2, row-1:row+2]).ravel()
print(nw, n, ne, w, _, e, sw, s, se)
# (None, None, None, None, 0, 1, None, 8, 9)
请注意
print(board)
时
这是值格式化的方式。所以也许你想要board[row-1:row+2, column-1:column+2]
。当然,您可以定义自己的print_board
函数,然后可以自由使用您喜欢的任何约定。