How To Give Array Input In Python