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

JS · 04 Jan 2021

리스트에서 첫번째와 마지막 요소를 가져오는 방법을 소개합니다.

  • list index로 첫번쨰, 마지막 요소 가져오기
  • list comprehension을 이용한 방법
  • list slicing을 이용한 방법

list index로 첫번쨰, 마지막 요소 가져오기

index를 이용하여 첫번째와 마지막 요소를 가져올 수 있습니다. list[0]은 리스트에서 첫번째, list[-1]은 리스트에서 마지막의 값을 가져옵니다.

my_list = [6, 2, 1, 8, 3]
result = [my_list[0], my_list[-1]]

print(my_list[0])
print(my_list[-1])
print(result)

Output:

6
3
[6, 3]

list comprehension을 이용한 방법

list comprehension을 이용할 수도 있습니다. for i in (0, -1)으로 첫번째와 마지막에만 접근하여 리스트로 만들 수 있습니다.

my_list = [6, 2, 1, 8, 3]
result = [my_list[i] for i in (0, -1)]

Output:

[6, 3]

List slicing을 이용한 방법

list slicing의 sytax는 list[start:stop:step]입니다. step을 list length - 1로 설정하면, 첫번째와 마지막 요소만 잘라서 가져오게 됩니다.

my_list = [6, 2, 1, 8, 3]
result = my_list[::len(my_list)-1]

Output:

[6, 3]
댓글을 보거나 쓰려면 이 버튼을 눌러주세요.
codechachaCopyright ©2019 codechacha