[python] 리스트의 크기와 값 한 번에 입력받기

5648: 역원소 정렬

input 10 5 2233
output 10, [5, 2223]

기존의 방식

numList = list(map(int, input().rsplit()))
numLen = numList[0]
numList.pop(0)

개선된 방식

numLen, *numList = input().rsplit()
numList = list(map(int, numList))

이게 가능한 이유는 python에서 *를 통해 argument를 따로 받을 수 있기 때문이다. **를 사용하면 key가 있는 값을 받을 수 있다.

def printArguemnts(value, *arg, **keyArg):
    print(value)
    print(arg)
    print(keyArg)


printArguemnts(1, 2, 3, 4, 5, name="Hej", color="yellow")

출력🔻

1
(2, 3, 4, 5)
{'name': 'Hej', 'color': 'yellow'}

함수의 인자로 받으면 *arg에서 tuple이, 아닐 때는 list가 나온다.

참고 자료

다른 사람 코드