Array Searching In Python Code