Programming/Python

Python list 원소 입력 : 입력 시 원소 증감하기

돌돌김 2021. 1. 5. 02:22

알고리즘 문제를 풀다보면 원소의 시작이 1부터 되는 경우가 종종 있다.

 

리스트에서 인덱스는 0부터 시작하므로, 입력을 받을 때 부터 1을 빼고 리스트에 저장하려면 lambda를 사용하면 된다

 

 

# 입력 예시 
# 1 2 3 4 5 6

# 기존 방식대로 입력을 받아서 1씩 빼주는 경우
mylist = list(map(int, input().split()))
    for i in range(len(mylist)):
        mylist[i] -= 1
print(*mylist) # 0 1 2 3 4 5

# lambda를 활용하여 입력 받을 때 부터 1을 빼줌
mylist = list(map(lambda x : int(x)-1, input().split()))
print(*mylist) # 0 1 2 3 4 5
 
1 2 3 4 5 6