質問
次のようなリストのリストがあります
[[1, 2, 3,],[4, 5, 6,],[7, 8, 9]]
.
グラフで表すと次のようになります。
1 2 3
4 5 6
7 8 9
セルの隣接セルの値を水平、垂直、斜めにチェックするエレガントなアプローチを探しています。たとえば、[0][2] の近傍は [0][1]、[1][1]、[1][2]、または数値 2、5、6 です。
今、私は、すべての値を順番にチェックするブルートフォース攻撃を実行できることに気付きました。
[i-1][j]
[i][j-1]
[i-1][j-1]
[i+1][j]
[i][j+1]
[i+1][j+1]
[i+1][j-1]
[i-1][j+1]
しかし、それは簡単です。もっと洗練されたアプローチをいくつか見ることで、さらに学ぶことができると思いました。
解決
# Size of "board"
X = 10
Y = 10
neighbors = lambda x, y : [(x2, y2) for x2 in range(x-1, x+2)
for y2 in range(y-1, y+2)
if (-1 < x <= X and
-1 < y <= Y and
(x != x2 or y != y2) and
(0 <= x2 <= X) and
(0 <= y2 <= Y))]
>>> print(neighbors(5, 5))
[(4, 4), (4, 5), (4, 6), (5, 4), (5, 6), (6, 4), (6, 5), (6, 6)]
これがクリーンと見なされるかどうかはわかりませんが、この1ライナーは、繰り返し処理を行い、エッジケースを破棄することで、すべてのネイバーを提供します。
他のヒント
mb ...
from itertools import product, starmap
x, y = (8, 13)
cells = starmap(lambda a,b: (x+a, y+b), product((0,-1,+1), (0,-1,+1)))
// [(8, 12), (8, 14), (7, 13), (7, 12), (7, 14), (9, 13), (9, 12), (9, 14)]
print(list(cells)[1:])
正方行列があると仮定します:
from itertools import product
size = 3
def neighbours(cell):
for c in product(*(range(n-1, n+2) for n in cell)):
if c != cell and all(0 <= n < size for n in c):
yield c
itertools.product
そして、Pythonの yield expression およびスター演算子、関数はかなり dry ですが、十分に読みやすくなっています。
マトリックスサイズが3の場合、(必要に応じて) list
でネイバーを収集できます:
>>> list(neighbours((2,2)))
[(1, 1), (1, 2), (2, 1)]
関数の機能は次のように視覚化できます:
for x_ in range(max(0,x-1),min(height,x+2)):
for y_ in range(max(0,y-1),min(width,y+2)):
if (x,y)==(x_,y_): continue
# do stuff with the neighbours
>>> a=[[1, 2, 3], [4, 5, 6], [7, 8, 9]]
>>> width=height=3
>>> x,y=0,2
>>> for x_ in range(max(0,x-1),min(height,x+2)):
... for y_ in range(max(0,y-1),min(width,y+2)):
... if (x,y)==(x_,y_): continue
... print a[x_][y_]
...
2
5
6
これを行うクリーンな方法はありません。本当に必要な場合は、関数を作成できます:
def top(matrix, x, y):
try:
return matrix[x][y - 1];
except IndexError:
return None
誰かが直接(非対角)隣人を選ぶ別の方法に興味があるなら、ここに行きます:
neighbors = [(x+a[0], y+a[1]) for a in
[(-1,0), (1,0), (0,-1), (0,1)]
if ( (0 <= x+a[0] < w) and (0 <= y+a[1] < h))]
ここにリストがあります:
(x - 1, y - 1) (x, y - 1) (x + 1, y - 1)
(x - 1, y) (x, y) (x + 1, y)
(x - 1, y + 1) (x, y + 1) (x + 1, y + 1)
つまり、(x、y)の水平方向の近傍は(x +/- 1、y)です。
垂直方向の近傍は(x、y +/- 1)です。
対角線上の近傍は(x +/- 1、y +/- 1)です。
これらの規則は、無限行列に適用されます。 初期値(x、y)がエッジにある場合、近傍が有限行列に収まるようにするには、近傍の座標にもう1つの制限、つまり行列サイズを適用します。
>>> import itertools
>>> def sl(lst, i, j):
il, iu = max(0, i-1), min(len(lst)-1, i+1)
jl, ju = max(0, j-1), min(len(lst[0])-1, j+1)
return (il, iu), (jl, ju)
>>> lst = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
>>> tup = 0, 2
>>> [lst[i][j] for i, j in itertools.product(*sl(lst, *tup)) if (i, j) != tup]
[2, 5, 6]
どのようにエレガントに見えるかわかりませんが、ハードコーディングなしで機能するようです。
これはすべてのインデックスを生成します:
def neighboring( array ):
nn,mm = len(array), len(array[0])
offset = (0,-1,1) # 0 first so the current cell is the first in the gen
indices = ( (i,j) for i in range(nn) for j in range(mm) )
for i,j in indices:
all_neigh = ( (i+x,j+y) for x in offset for y in offset )
valid = ( (i,j) for i,j in all_neigh if (0<=i<nn) and (0<=j<mm) ) # -1 is a valid index in normal lists, but not here so throw it out
yield valid.next(), valid ## first is the current cell, next are the neightbors
for (x,y), neigh in neighboring( l ):
print l[x][y], [l[x][y] for x,y in neigh]
多分あなたは数独ボックスをチェックしています。ボックスがn x nで、現在のセルが(x、y)の場合、チェックを開始します。
startingRow = x / n * n;
startingCol = y/ n * n
おかげで @JS_is_bad 隣人についての素晴らしいヒントが得られます。この問題の実行コードは次のとおりです
def findNeighbours(l,elem):
#This try is for escaping from unbound error that happens
#when we try to iterate through indices that are not in array
try:
#Iterate through each item of multidimensional array using enumerate
for row,i in enumerate(l):
try:
#Identifying the column index of the givem element
column=i.index(elem)
except ValueError:
continue
x,y=row,column
# hn=list(((x,y+1),(x,y-1))) #horizontal neighbours=(x,y+/-1)
# vn=list(((x+1,y),(x-1,y))) #vertical neighbours=(x+/-1,y)
# dn=list(((x+1,y+1),(x-1,y-1),(x+1,y-1),(x-1,y+1))) #diagonal neighbours=(x+/-1,y+/-1)
#Creating a list with values that are actual neighbors for the extracted index of array
neighbours=[(x,y+1),(x,y-1),(x+1,y),(x-1,y),(x+1,y+1),(x-1,y-1),(x+1,y-1),(x-1,y+1)]
#Creating a universe of indices from given array
index_list=[(i,j) for i in range(len(l)) for j in range(len(l[i]))]
#Looping through index_list and nested loop for neighbours but filter for matched ones
# and extract the value of respective index
return_values=[l[index[0]][index[1]] for index in index_list for neighbour in neighbours if index==neighbour]
return return_values,neighbours
except UnboundLocalError:
return []
ここでラムダが気に入らなかったとしても、ラムダはコードをきれいに見せます。@ johniek_compには非常にきれいなソリューションTBHがあります
k,l=(2,3)
x = (0,-1,+1)
y = (0,-1,+1)
cell_u = ((k+a,l+b) for a in x for b in y)
print(list(cell_u))