How to make an array in python