알아두면 좋은 python 함수 - lambda, assert, map, filter

1. lambda()

def square(x):
	return x*x
    
     
 square = lambda x: x*x

 

books = [
	"신데렐라, 2008",
    "백설공주, 2007",
    "인어공주, 2020"
]

#Example 1
def get_title(row):
	split = row.split(',')
    return split[0]
sorted(books, key = get_title)


#Example 2
get_title = lambda row: row.split(',')[0]
sorted(books, key = get_title)


#Example 3
sorted(books, key = lambda row:row.split(',')[0])

 

• Example 1,2,3 은 모두 같은 일을 수행
• lambda를 이용해 코드를 짧게 구현 가능

2. assert()

• 두 가지의 값이 동일한지 확인하고 싶을 때 사용
• 두 값이 같으면 통과, 아니면 에러

def square1(x):
	return x*x
square2 = lambda x:x*x

assert(square1(3) == square(3))

 

• square1 과 square2가 동일한 기능을 수행하는지 확인 
• 두 값이 같지 않았을 경우에 assertionError 발생

 

3. map()

• 장점: 리스트 보다 빠름, 필요없는 연산은 하지 않음

books = [
	"신데렐라,Cinderella, 2008",
    "백설공주, Snow White, 2007",
    "인어공주, The Little Mermaid, 2020"
]

#Example 1
def get_eng_title(row):
	split = row.split(',')
    return split[1]
eng_titles = [get_eng_title(row) for row in books]

#Example 2 : map 사용
def get_eng_title(row):
	split = row.split(',')
    return split[1]
eng_titles = map(get_eng_title, books) #books 리스트에 속한 모든 원소들에 get_eng_title 함수 반영


#Example 3 : lambda, map 사용
eng_titles = map(lamda row: row.split(',')[1], books)

 

• Example2 에서 eng_titles 출력시, <map object at ~~~> 라고 뜬다(
  (map이라는 타입을 출력, 필요없는 연산을 하지 않은 map함수의 특징 때문)
• eng_titles[0] 이런 식으로 출력해야 값이 출력 됨
• 리스트 형태로 바꾸면 결과물 조회 가능 : list(eng_titles)

 

4. filter()

def starts_with_a(word):
	return word.startswith('r')
    
words = ['rabbit','dog','cat', 'red']
r_words = filter(starts_with_a, words)

 

• print(r_word) 는 filter 타입을 출력
• print(list(r_word) 형태로 바꾸면 결과물 조회 가능