You can initially start with setting the count count = -grid[row][col] Then you don't need to test for it at the end. Generally it would have been easier to just sum the values instead of testing them for 1 and then adding 1 to the counter More

I would have expected lists of booleans in a cleaned version of the map. More