How To Define Array Size In Python Formatting Output Python