initialize arrays in python