[python] How to declare array of zeros in python (or an array of a certain size)