2의 영역

정수 배열 arr가 주어집니다.
배열 안의 2가 모두 포함된
가장 작은 연속된 부분 배열을
return 하는 solution 함수를 완성해 주세요.

단, arr에 2가 없는 경우 [-1]을 return 합니다.

내 코드

def solution(arr):
    answer = []
    myString = ''.join(map(str,arr))
    myString = myString.replace('10','A')
    head = myString.find('2')
    tail = myString.rfind('2')
    if head == -1:
        answer = [-1]
    else:
        print(tail)
        answer = arr[head:tail+1]
    return answer

예시 코드

def solution(arr):
    if 2 not in arr:
        return [-1]
    return arr[arr.index(2) : len(arr) - arr[::-1].index(2)]