Computer >> 컴퓨터 >  >> 프로그래밍 >> Python

Python 리스트에서 첫 번째 요소와 마지막 요소 가져오는 방법

Python 프로그래밍을 하다 보면 리스트의 첫 번째 요소와 마지막 요소를 가져와야 하는 상황이 자주 발생합니다. 이때 주의할 점은 리스트의 길이를 고려하면서 원하는 요소에 접근해야 한다는 것입니다. 다행히 Python에서는 인덱스를 활용하는 다양한 방법으로 이 작업을 간단하게 처리할 수 있습니다. 아래에서 대표적인 세 가지 방법을 살펴보겠습니다.

1. 인덱스만 사용하기

Python 리스트에서 첫 번째 요소는 항상 인덱스 0에 할당되고, 마지막 요소는 음수 인덱스 -1로 접근할 수 있습니다. 따라서 이 두 인덱스 값을 리스트에 직접 적용하면 원하는 결과를 손쉽게 얻을 수 있습니다. 가장 직관적이고 널리 사용되는 방법입니다.

예제

Alist = ['Sun','Mon','Tue','Wed','Thu']
print("주어진 리스트 : ",Alist)

print("리스트의 첫 번째 요소 : ",Alist[0])
print("리스트의 마지막 요소 : ",Alist[-1])

실행 결과

위 코드를 실행하면 다음과 같은 결과가 출력됩니다.

주어진 리스트 : ['Sun', 'Mon', 'Tue', 'Wed', 'Thu']
리스트의 첫 번째 요소 : Sun
리스트의 마지막 요소 : Thu

2. 리스트 슬라이싱 활용하기

슬라이싱(slicing) 기법을 사용하면 콜론(:)을 이용해 요소의 위치를 직접 지정할 수 있습니다. 첫 번째 콜론 앞을 비워두면 처음부터 시작하고, 스텝(step) 값으로 len(Alist)-1을 지정하면 첫 번째 요소와 마지막 요소를 동시에 추출할 수 있습니다. 이 방법은 두 요소를 새로운 리스트 형태로 한 번에 얻고 싶을 때 유용합니다.

예제

Alist = ['Sun','Mon','Tue','Wed','Thu']
print("주어진 리스트 : ",Alist)

first_last = Alist[::len(Alist)-1]
print(first_last)

실행 결과

위 코드를 실행하면 다음과 같은 결과가 출력됩니다.

주어진 리스트 : ['Sun', 'Mon', 'Tue', 'Wed', 'Thu']
['Sun', 'Thu']

3. for 루프와 리스트 컴프리헨션 사용하기

for 루프와 in 연산자를 조합하여 인덱스 값 0과 -1을 순회하는 방법도 있습니다. 리스트 컴프리헨션(list comprehension)을 활용하면 코드를 한 줄로 간결하게 작성할 수 있어 가독성이 뛰어납니다.

예제

Alist = ['Sun','Mon','Tue','Wed','Thu']
print("주어진 리스트 : ",Alist)

first_last = [Alist[n] for n in (0,-1)]
print(first_last)

실행 결과

위 코드를 실행하면 다음과 같은 결과가 출력됩니다.

주어진 리스트 : ['Sun', 'Mon', 'Tue', 'Wed', 'Thu']
['Sun', 'Thu']

마무리

정리하면, 단일 요소만 필요하다면 Alist[0]Alist[-1]처럼 인덱스를 직접 사용하는 것이 가장 효율적이고, 첫 번째와 마지막 요소를 하나의 리스트로 묶어야 한다면 슬라이싱이나 리스트 컴프리헨션을 활용하는 것이 좋습니다. 상황에 맞는 방법을 선택해 코드의 가독성과 효율성을 높여보세요.