2D 배열의 행과 열에서 값 찾기

Question

지정된 값이 2D 배열의 주어진 행이나 주어진 열에 있는지 알아 내려고하고 있습니다.

함수 existsInRowOrCol 명백하게 혼합 형태의 중첩 된 목록으로 잘 작동하지 않는다 연산자를 이용한다.

어떻게 해결할 수 있습니까?

import numpy as np 

field = np.array([['.','.','.',8,'.','.',3,'.',4],  #0 
       ['.','.',6,'.','.','.','.',7,'.'],  #1 
       [2,'.','.','.',1,'.','.','.','.'],  #2 
       ['.',8,'.',3,'.',4,'.',6,'.'],   #3 
       [1,'.','.','.',5,'.','.',9,'.'],  #4 
       [5,'.','.','.',8,'.',7,'.','.'],  #5 
       ['.',9,'.',1,'.',8,'.',5,'.'],   #6 
       [3,'.','.',6,'.','.',8,'.','.'],  #7 
       ['.','.',8,'.',2,7,'.',1,'.'],   #8 
       ]) 

def existsInRowOrCol(entry, r, c): 
    row = field[r,:] 
    print(row) 
    if entry in row: 
     return True 
    col = field[:,c] 
    print(col) 
    if entry in col: 
     return True 
    return False 

print(existsInRowOrCol(8, 0, 0)) 

마지막 명령은 false를 출력하지만 true 여야합니다.

참고 : 점을 0으로 바꾸면 작동합니다. 당신이 나를 위해 일한 따옴표로 항목을 갖는 괜찮다 경우

Answer 1

:

>>> a = numpy.array(['.','.','8']) 
>>> a 
array(['.', '.', '8'], 
    dtype='|S1') 
>>> row = a[:] 
>>> entry = 8 
>>> if entry in row: 
... print 'true' 
... 
>>> entry = '8' 
>>> if entry in row: 
... print 'true' 
... 
true 

그냥 당신이 따옴표 항목을 통과해야합니다.