Product of Array Except Self - LeetCode

[@arsenypogosov7206]

ans = [1] * len(nums)
prod, zero_count = 1, 0
for i in nums:
if i != 0:
prod *= i
else:
zero_count += 1
for i in range(len(nums)):
ans[i] = 0 if zero_count - (nums[i] == 0) > 0 else prod // (nums[i] if nums[i] != 0 else 1)
return ans