[Python] 코딩테스트 백업용

 

 

리스트 평균값 구하기

def solution(numbers):
    sum = 0
    for i in numbers:
        sum += i
    answer = sum / len(numbers)
    print(answer)
    return answer

 

더 간단한 방법

def solution(numbers):
    return sum(numbers) / len(numbers)

 

numpy 사용

import numpy as np
def solution(numbers):
    return np.mean(numbers)

 

---

짝수 홀수 갯수 구하기

 

나의 풀이

def solution(num_list):
    list1 = [n for n in num_list if n % 2 == 0]
    list2 = [n for n in num_list if n % 2 != 0]
    answer = [len(list1), len(list2)]
    return answer

 

기발한 풀이..

def solution(num_list):
    answer = [0,0]
    for n in num_list:
        answer[n%2]+=1
    return answer

---

나머지 구하기

def solution(money):
    result, ect = money // 5500, money % 5500
    return [result, ect]

 

python divmode() 함수 사용

이 함수는 몫과 나머지를 [몫, 나머지] 형태로 리스트에 담아서 리턴한다.

def solution(money):
    return divmod(money, 5500)

---

배열의 유사도

 

나의 풀이(for문 사용)

def solution(s1, s2):
    answer = 0
    for n in s1:
        for m in s2:
            if n == m:
                answer += 1
    return answer

 

set 사용(교집합 구한 후 갯수 세기)

def solution(s1, s2):
    return len(set(s1)&set(s2));

---