파이썬으로 문자열에서 단어 등장 횟수 계산하기

이번 튜토리얼에서는 문자열 안에서 특정 단어가 몇 번 등장하는지 세는 프로그램을 작성해 보겠습니다. 문자열과 찾으려는 단어가 주어지면, 해당 단어가 문자열에 나타나는 빈도를 계산하면 됩니다.

예를 들어, I am a programmer. I am a student.라는 문자열과 단어 am이 주어졌다고 가정해 봅시다. 'am'이라는 단어가 문자열 안에 두 번 등장하므로, 우리가 작성할 프로그램은 숫자 2를 반환해야 합니다.

아래의 단계를 따라 목표를 달성해 봅시다.

알고리즘

1. 문자열과 단어를 각각 변수로 초기화합니다.
2. split() 메서드를 사용해 문자열을 공백 기준으로 분리합니다. 그러면 단어들의 리스트를 얻게 됩니다.
3. 카운트 변수(count)를 0으로 초기화합니다.
4. 리스트를 반복(iteration)합니다.
4.1. 리스트의 단어가 주어진 단어와 일치하는지 확인합니다.
4.1.1. 두 단어가 일치하면 카운트를 1 증가시킵니다.
5. 카운트를 출력합니다.

먼저 스스로 코드를 작성해 본 뒤, 아래의 예제와 비교해 보세요.

예제 코드

## 문자열과 단어 초기화
string = "I am programmer. I am student."
word = "am"
## 공백 기준으로 문자열 분리
words = string.split()
## 카운트 변수를 0으로 초기화
count = 0
## 리스트를 반복하며 단어 확인
for w in words:
    ## 단어 일치 여부 검사
    if w == word:
        ## 일치하면 카운트 증가
        count += 1
## 카운트 출력
print(count)

실행 결과

위 프로그램을 실행하면 다음과 같은 결과가 출력됩니다.

2

마무리

이 프로그램의 핵심은 split() 메서드로 문자열을 단어별로 분리한 뒤, 반복문으로 각 단어를 비교하는 것입니다. 참고로 이 방법은 대소문자를 구분합니다. 대소문자를 무시하고 싶다면 split() 호출 전에 lower() 메서드로 문자열 전체를 소문자로 변환한 후 비교하면 됩니다.

프로그램과 관련해 궁금한 점이 있다면 댓글로 남겨주세요.