[알고리즘] Python 기초.3

 

def solution(array, commands):
    return list(map(lambda x:sorted(array[x[0]-1:x[1]])[x[2]-1], commands))

append

append를 사전에서 검색해 보면 "덧붙이다, 첨부하다"라는 뜻이 있다. 이 뜻을 안다면 다음 예가 바로 이해될 것이다. append(x)는 리스트의 맨 마지막에 x를 추가하는 함수이다.

>>> a = [1, 2, 3]
>>> a.append(4)
>>> a [1, 2, 3, 4]

리스트 안에는 어떤 자료형도 추가할 수 있다.

다음 예는 리스트에 다시 리스트를 추가한 결과이다.

>>> a.append([5,6])
>>> a [1, 2, 3, 4, [5, 6]]

 

sorted vs sort

일단 sorted 함수는 정렬된 새로운 리스트를 리턴시켜줍니다. 반면에 sort 메소드는 아무것도 리턴시켜주지 않습니다(None을 리턴시켜줌).

some_list = [5, 7, 2, 3, 1]

print(sorted(some_list))
print(some_list.sort())

[1, 2, 3, 5, 7]
None

sorted

some_list = [5, 7, 2, 3, 1]

sorted(some_list)
print(some_list)

[5, 7, 2, 3, 1]

sort

some_list = [5, 7, 2, 3, 1]

some_list.sort()
print(some_list)

[1, 2, 3, 5, 7]

map

def func(x):
	return x * 2

 

인자로 받은 정수를 두배로 곱하여 반환해 주는 매우 간단한 함수이다.

이 함수에 인자를 map() 함수를 이용해 전달해 보자.

 

>>> map( func, [1, 2, 3, 4] )

[2, 4, 6, 8]

 

 위와 같이 map() 함수는 for문과 같은 반복문을 사용하지 않아도 지정한 함수로 인자를 여러번 전달해 그 결과를 list 형태로 뽑아 주는 유용한 함수이다. 한줄로 처리되다 보니 매우 간결한 형태의 코딩이 가능하다는 것이 큰 장점이다.

 

 map() 함수의 경우 보통 위와 같이 인자를 list 형태로 전달하는게 일반적이지만 iterable 한 형태인 dictionary 같은 인자도 가능하다. 

 

>>> x = { 1 : 10, 2 : 20, 3: 30 }
>>> map(func, x)

[ 2, 4, 6 ]

 

lambda

32.1.1  람다 표현식 자체를 호출하기

람다 표현식은 변수에 할당하지 않고 람다 표현식 자체를 바로 호출할 수 있습니다. 다음과 같이 람다 표현식을 ( )(괄호)로 묶은 뒤에 다시 ( )를 붙이고 인수를 넣어서 호출하면 됩니다.

• (lambda 매개변수들: 식)(인수들)

>>> (lambda x: x + 10)(1)

11

32.1.2  람다 표현식 안에서는 변수를 만들 수 없다

람다 표현식에서 주의할 점은 람다 표현식 안에서는 새 변수를 만들 수 없다는 점입니다. 따라서 반환값 부분은 변수 없이 식 한 줄로 표현할 수 있어야 합니다. 변수가 필요한 코드일 경우에는 def로 함수를 작성하는 것이 좋습니다.

>>> (lambda x: y = 10; x + y)(1)

SyntaxError: invalid syntax

단, 람다 표현식 바깥에 있는 변수는 사용할 수 있습니다. 다음은 매개변수 x와 람다 표현식 바깥에 있는 변수 y를 더해서 반환합니다.

>>> y = 10
>>> (lambda x: x + y)(1)

11

 

32.1.3  람다 표현식을 인수로 사용하기

람다 표현식을 사용하는 이유는 함수의 인수 부분에서 간단하게 함수를 만들기 위해서 입니다. 이런 방식으로 사용하는 대표적인 예가 map입니다.

람다 표현식을 사용하기 전에 먼저 def로 함수를 만들어서 map을 사용해보겠습니다. 다음과 같이 숫자를 받은 뒤 10을 더해서 반환하는 함수 plus_ten을 작성합니다. 그리고 map에 plus_ten 함수와 리스트 [1, 2, 3]을 넣습니다. 물론 map의 결과는 map 객체이므로 눈으로 확인할 수 있도록 list를 사용해서 리스트로 변환해줍니다.

>>> def plus_ten(x): 
...		return x + 10
...
    
>>> list(map(plus_ten, [1, 2, 3]))

[11, 12, 13]

plus_ten 함수는 매개변수 x에 10을 더해서 반환하므로 리스트 [1, 2, 3]이 [11, 12, 13]으로 바뀌었습니다. 지금까지 map을 사용할 때 map(str, [1, 2, 3])와 같이 자료형 int, float, str 등을 넣었죠? 사실 plus_ten처럼 함수를 직접 만들어서 넣어도 됩니다.

이제 람다 표현식으로 함수를 만들어서 map에 넣어보겠습니다.

>>> list(map(lambda x: x + 10, [1, 2, 3]))

[11, 12, 13]

plus_ten 함수 대신 람다 표현식 lambda x: x + 10을 넣었습니다. 전체적으로 보면 코드가 세 줄에서 한 줄로 줄었죠? 이처럼 람다 표현식은 함수를 다른 함수의 인수로 넣을 때 매우 편리합니다.

 

참고 | 람다 표현식으로 매개변수가 없는 함수 만들기

람다 표현식으로 매개변수가 없는 함수를 만들 때는 lambda 뒤에 아무것도 지정하지 않고 :(콜론)을 붙입니다. 단, 콜론 뒤에는 반드시 반환할 값이 있어야 합니다. 왜냐하면 표현식(expression)은 반드시 값으로 평가되어야 하기 때문입니다.

>>> (lambda : 1)()

1

>>> x = 10
>>> (lambda : x)()

10