오늘 시간적 여유를 가지고 만든 고급 그래프 (Y축 2개, Plot 3개)

[sl100104]

구글과 chatGPT의 도움으로 인내 끝에 만든 Plot이다. 여러분들도 코드를 save 해 두었다가 훗날 유용하게 쓰시길 바란다,

import numpy as np
import matplotlib.pyplot as plt

x = np.arange(0, 15, 0.2)
data_1 = np.sin(x)
data_2 = np.cos(x)
data_3 = np.cos(x**2)

fig, ax1 = plt.subplots()
ax1.set_xlabel('X-axis')
ax1.set_ylabel('Y1-axis', color='red')
ax1.plot(x, data_1, color='black', label='sin(x)')
ax1.plot(x, data_2, color='green',  label='cos(x)')  
ax1.tick_params(axis='y', labelcolor='red')

ax2 = ax1.twinx()
ax2.set_ylabel('Y2-axis', color='blue')
ax2.plot(x, data_3, color='blue', label='cos(x^2)')  
ax2.tick_params(axis='y', labelcolor='blue')
ax2.set_ylim(-1, 5)  # y2축의 범위를 -1에서 5로 설정

# 범례 추가
lines, labels = ax1.get_legend_handles_labels()
lines2, labels2 = ax2.get_legend_handles_labels()
ax2.legend(lines + lines2, labels + labels2, loc="upper right")
plt.show()

# 그림 크기 조정
fig.set_size_inches(6, 4)

# DPI 설정하여 저장
plt.savefig("myplot2.png", dpi=300)
plt.close(fig)