Define An Array Of Size N In Python