Python Fill Array With Zeros Of A Function