Number of Laser Beams in a Bank LeetCode solution in Python

1
class Solution(object):
2
    def numberOfBeams(self, bank):
3
        ans, pre = 0, 0
4
        for s in bank:
5
            n = s.count("1")
6
            if n == 0:
7
                continue
8
            ans += pre * n
9
            pre = n
10
        return ans