[Programmers] LV 0. 짝수 홀수 개수 (Python/파이썬)

▶ 문제 설명

정수가 담긴 리스트 num_list가 주어질 때, num_list의 원소 중 짝수와 홀수의 개수를 담은 배열을 return 하도록 solution 함수를 완성해보세요.

---

▶ 제한 사항

• 1 ≤ num_list의 길이 ≤ 100
• 0 ≤ num_list의 원소 ≤ 1,000

---

▶ 입출력 예

num_list	result
[1, 2, 3, 4, 5]	[2, 3]
[1, 3, 5, 7]	[0, 4]

---

▶입출력 예 설명

입출력 예 #1

• [1, 2, 3, 4, 5]에는 짝수가 2, 4로 두 개, 홀수가 1, 3, 5로 세 개 있습니다.

입출력 예 #2

• [1, 3, 5, 7]에는 짝수가 없고 홀수가 네 개 있습니다.

---

▶ Code

def solution(num_list):
    a = []
    b = []
    for i in range(len(num_list)):
        if (num_list[i] % 2) == 0:
            a.append(i)
        else:
            b.append(i)
    answer = [len(a), len(b)]
    return answer

▶ Point

변수 a와 b를 list 형식으로 만들기 위해 '[ ]' 으로 초기 설정해놓는다.

 

for 문에서 i를 num_list의 길이 만큼 반복시키기 위해 'for i in range(len(num_list)):'로 작성해놓고, 변수 a 에 num_list의 i번째 원소를 2로 나눈 나머지가 0인 짝수인 경우 해당 원소를 추가하여 저장한다. 마찬가지로 변수 b에 num_list의 i번째 원소를 2로 나눈 나머지가 0이 아닌 홀수를 추가하여 저장한다.

 

이후 answer라는 list에 변수 a의 갯수와 변수 b의 갯수를 입력하기 위해 'answer = [len(a), len(b)]' 으로 작성하여 출력한다.