그럼 graph_plot() 함수를 작성해 봅시다. 앞에서 원래 코드를 함수로 구성할 때 graph_plot()의 매개변수는 총 3개였습니다. 각각 인구 데이터, 그래프 라벨, 그래프 제목을 의미하는 popu_list, label_list, graph_title였죠. 프로그램에서 graph_plot() 함수를 호출할 때 popu_list에는 이차원 리스트를, label_list에는 일차원 리스트를, graph_title에는 문자를 입력하는 함수를 작성합니다.

import matplotlib.pyplot as plt

def graph_plot(popu_list, label_list, graph_title):
    plt.rc('font', family='Malgun Gothic')
    plt.title(graph_title)
    for i in range(len(popu_list)):
        plt.plot(range(24), popu_list[i], label=label_list[i])
    plt.legend()
    plt.xlabel('시간대')
    plt.ylabel('평균인구수')
    plt.xticks(range(24), range(24))
    plt.show()

한글 글꼴을 설정하고, 전달받은 매개변수 graph_title로 그래프 제목을 넣었습니다. 그 다음 인구 데이터로 입력받은 popu_list의 첫 행부터 마지막 행까지 반복합니다. 그리고 반복문 내부에서 popu_list의 인덱스와 label_list의 인덱스를 맞춰 plt.plot()으로 그래프를 그립니다. 그래프 색상은 자동으로 설정되도록 따로 지정하지 않았습니다. 나머지 공통 부분을 추가해서 함수를 완성했습니다.