[Python] count() 함수

count 함수를 설명하기 위해 아래 백준 2577번 숫자의 개수 문제를 가져와봤습니다!

 

count함수는 문자열, 리스트 안에서 찾고싶은 문자의 개수를 찾을 수 있습니다.

 

문자열의 경우를 예로 들면,

a = 'i am kimdongjun'
print(a.count('i'))

2

문자열에서 i의 개수를 찾아 반환하였습니다.

 

리스트의 경우를 예로 들면,

a = [1, 5, 3, 2, 1]
print(a.count(1))

2

리스트 a에서 1의 개수를 찾아 반환하였습니다.

 

※ dictionary, set자료형에서는 count함수를 사용할 수 없습니다!

 

그럼 아래의 문제에서 count를 사용하면 편리하겠죠?

 

 

 

먼저 위 문제를 count함수를 사용하지 않고 풀어보면 아래와 같이 코드가 상대적으로 길어집니다.

a = int(input())
b = int(input())
c = int(input())
cnt = 0

x = list(str(a*b*c))

for i in range(0, 10):
    for j in range(0, len(x)):
        if x[j] == str(i):
            cnt = cnt + 1
    print(cnt)
    cnt = 0

 

이중 for문을 사용해서 하나하나 비교하고 같은 숫자이면 카운트하는 방식으로 풀이했습니다. 

count함수를 사용하면 더 간편해질텐데요,

a = int(input())
b = int(input())
c = int(input())
cnt = 0

x = list(str(a*b*c))

for i in range(0, 10):
    cnt = x.count(str(i))
    print(cnt)

 

이처럼 파이썬의 다양한 함수를 알아두면 문제풀이가 더 간단해집니다.