How To Define Array Input In Python