조건(특정 문자열과 연관있는 문자의 빈도수 세기)

Question

from collections import Counter

def find_related_groceries(grocery):
    f = open('Assignment_text.txt', 'r')
    all_food_list = {}
    for line in f:
        year_month = line.split(',')[0]
        if all_food_list.get(year_month) is None:
            all_food_list[year_month] = []

        groceries = [v for v in line.split(",")[1:-1] if v]

        for grocery_item in groceries:
            all_food_list[year_month].append(grocery_item)

    flag = False
    for year_month in all_food_list:
        cnt = Counter(all_food_list[year_month])
        mc = cnt.most_common(n=3)

        all_food_list[year_month] = {}

        for item in mc:
            all_food_list[year_month][item[0]] = item[1]
            if item[0] == grocery:
                flag = True

    f.close()
    if flag:
        return all_food_list
    else:
        return None

print(find_related_groceries('coffee'))

 

 

첨부한 형태의 text 파일에서 coffee 와 관련있는 (같이 판매된) grocery를 딕셔너리 형태로 출력해주는 코드를 짜보았는데요. 저는 계속 none 만 출력됩니다ㅠㅜㅠ