How To Give Array Input In Python Definition