[Python] 리스트에서 원하는 요소만 추출하기

리스트에서 홀수인 데이터만 출력하도록 하는 함수

#주어진 리스트
num_list = {1, 5, 7, 15, 16, 22, 28, 29}

def get_odd_num(num_list):
    # TODO

print(get_odd_num(num_list))

#출력
[1, 5, 7, 15, 29]

---

ⓐ 리스트 자체를 파라미터로 받아오는 함수

# 방법 1

#주어진 리스트
num_list = {1, 5, 7, 15, 16, 22, 28, 29}

def get_odd_num(num_list):
    return [x for x in num_list if(x % 2)]

print(get_odd_num(num_list))

# 방법 2

#주어진 리스트
num_list = [1, 5, 7, 15, 16, 22, 28, 29]

def get_odd_num(num_list):
    odd_list = []

    for x in num_list:
        if(x % 2 == 1):
            odd_list.append(x)

    return odd_list

print(get_odd_num(num_list))

#출력
[1, 5, 7, 15, 29]

---

*list comprehension 리스트 컴프리헨션:

[(변수를 활용한 값) for (사용할 변수 이름) in (순회할 수 있는 값 or 필터링할 수 있는 값(조건문 ...)]

+ 참고 <list comprehension>

---

ⓑ 파라미터로 원소를 받는 경우

# 방법 1

#주어진 리스트
num_list = {1, 5, 7, 15, 16, 22, 28, 29}

odd_list = [x for x in num_list if x % 2 == 0]

print(odd_list)

# 방법 2

#주어진 리스트
num_list = {1, 5, 7, 15, 16, 22, 28, 29}

def get_odd(n):
    return [n % 2 == 1]

odd_list = list(filter(get_odd, num_list))

print(odd_list)

---

*filter 함수:

filter(함수이름, 원본 리스트)

return 값: iterator(filter object)

원본 리스트의 요소들을 하나하나 함수에 넣어서 반환이 true인 값을 다시 리스트로 만들어준다.

 

filter 함수의 인자로 들어가는 함수의 리턴 타입은 bool형이어야 한다.

+ 참고 <filter 함수> https://helloworld-fromuser.tistory.com/19