Define Array Of Size N In Python