[python] 행이 행에서 몇 번 발생하는지 계산하기 (numpy)



Answers

Question

주어진 행이 numpy 2D 배열로 나타나는 횟수를 세는 더 좋은 방법이 있습니까?

def get_count(array_2d, row):
    count = 0
    # iterate over rows, compare
    for r in array_2d[:,]:
        if np.equal(r, row).all():
            count += 1
    return count    

# let's make sure it works

array_2d = np.array([[1,2], [3,4]])
row = np.array([1,2])       

count = get_count(array_2d, row)
assert(count == 1)



Links



Tags

python python   numpy