플로팅을 위한 사용자 정의 함수 업데이트하기

이전에 연습에서 사용했던 플롯 생성 함수를 떠올려 보세요:

def make_plot(dataset, ratio, comp_type):
  whole_dat = []
  for industry in comp_type:
    dat = dataset.loc[dataset["comp_type"]==industry]
    dat_avg = dat.pivot_table(index="Year",
                              values=ratio).reset_index()
    dat_avg["company"] = f"Avg {type}"
    dat_avg["comp_type"] = industry
    whole_dat.append(pd.concat([dat,
                                dat_avg]))

  plot_df = pd.concat(whole_dat).reset_index(drop=True)
  sns.relplot(data=plot_df,
              x="Year",
              y="gross_margin",
              hue="company",
              col="comp_type",
              kind="line")
  plt.show()
  plt.close()

이 함수는 x축에 연도만 두는 선 그래프만 만들 수 있다는 점에 주목하세요. 이 연습에서는 이 함수의 업데이트된 버전을 알아봅니다.