How To Give Array As Input In Python