Python 2D数组处理

Question

I have an input array of the form

1 0 0 0 1
1 1 1 0 0
1 0 0 0 0
0 0 0 1 1
0 1 0 0 0

I want to calculate the density map and get output in the form (summing up all the values around a value including the value)

3 4 2 2 1
4 5 2 2 1
3 4 3 3 2
2 2 2 2 2
1 1 2 2 2

import numpy as np

def get_value(arr,i,j):
    """
    This function will return the value at the index if present. If list index out of range, it will return 0
    """
    try:
        return arr[i][j]
    except:
        return 0

#input array
arr = [[1, 0, 0, 0, 1],
 [1, 1, 1, 0, 0],
 [1, 0, 0, 0, 0],
 [0, 0, 0, 1, 1],
 [0, 1, 0, 0, 0]]

np_arr = np.array(arr) # converting to numpy array

r =1 # initializing the radius r as 1
n =5 # initializing the size of the input matrix as 5
np_arr_result = np.zeros(shape=(5,5)) # initializing the output array

for i in range(0,5):
    for j in range(0,5):
        np_arr_result[i][j] = get_value(np_arr,i,j) + get_value(np_arr,i-r,j-r) + get_value(np_arr,i-r,j)  + get_value(np_arr,i-r,j+r) + get_value(np_arr,i,j-r)  + get_value(np_arr, i, j+r) + get_value(np_arr, i+r, j-r) + get_value(np_arr, i+r, j) + get_value(np_arr, i+r, j+r)

print("output")
print(np_arr_result)

However, I am not getting the right output and getting the result as:

output
[[ 5.  5.  3.  2.  1.]
 [ 5.  5.  2.  2.  1.]
 [ 4.  4.  3.  3.  2.]
 [ 3.  2.  2.  2.  2.]
 [ 2.  1.  2.  2.  2.]]

What could have gone wrong?

Answer 1

This is a perfect task for a 2D convolution:

data = numpy.array([
    [1, 0, 0, 0, 1],
    [1, 1, 1, 0, 0],
    [1, 0, 0, 0, 0],
    [0, 0, 0, 1, 1],
    [0, 1, 0, 0, 0],
])

kernel = numpy.ones((3, 3), dtype=data.dtype)
# array([[1, 1, 1],
#        [1, 1, 1],
#        [1, 1, 1]])

scipy.signal.convolve2d(data, kernel, mode='same')
# array([[3, 4, 2, 2, 1],
#        [4, 5, 2, 2, 1],
#        [3, 4, 3, 3, 2],
#        [2, 2, 2, 2, 2],
#        [1, 1, 2, 2, 2]])

Answer 2

That is because for numpy, the index -1 is for the last element, so it will not assert IndexError.

just modify the get_value() function as :

def get_value(arr,i,j):
    """
    This function will return the value at the index if present. If list index out of range, it will return 0
    """
    if (i>-1 and j >-1 and i < arr.shape[0] and j < arr.shape[1]):
        return arr[i][j]
    else:
        return 0