[Python] 그래프 한번에 여러개 그리(plt.subplots())

안녕하세요.

집근처 개발자 입니다.

이번엔 그래프를 그리는데 한번에 여러개의 그래프를 표현하는 방법을 알려드릴게요.

여러개의 Legend가 생기는 것이 아니고 말그대로 여러개의 sub 그래프를 생성하는 방법입니다.

아래 사진과 같은 그래프를 생성하는 방법입니다.

size_graph = 5
for idx_x in range(size_graph):
    fig, axes = plt.subplots(1, 2, figsize=(25, 3))
    axes[0].plot(np.random.rand(100))
    axes[0].set_ylim(0, 1.2)
    axes[0].set_ylabel('Probability', fontsize=20)
    axes[0].grid()
    
    axes[1].plot(np.random.rand(100))
    axes[1].set_ylim(0, 1.2)
    axes[1].set_ylabel('Probability', fontsize=20)
    axes[1].grid()
    
    for ax in fig.axes:
        plt.sca(ax)
        plt.tick_params(axis='both', which='major', labelsize=15)
        plt.legend(fontsize=14)        

코드를 보시면 몇 row를 생성할 것인지를 size_graph로 설정할 수 있습니다,

설정한 size_graph의 크기 만큼 loop을 돌여 graph row를 생성하는 코드 입니다.

추가 설명이 필요해 보이는 부분의 추가 설명입니다.

grid(): x, y축 구분선

set_ylim: y축의 크기 설정

set_xlim: x축의 크기 설정

set_ylabel: y축의 label을 설정하고, fontsize로 글자 크기 설정 가능

plt.tick_params: x,y축의 scale 표시 값의 크기를 변경 설정 가능

간단합니다.