[ Matplotlib ] 특정 값만 색상 다르게 지정하기

본문으로

안녕하세요. 녀기입니다.

 

얼마 전에 빅분기 필기를 쳤습니다.

 

정말 빅분기를 공부할 때도 느꼈지만,

 

내용이 너무 난잡해요ㅠㅠ

 

여기저기 흩어져 있는 내용을 다 짜깁기 해서 시험을 보는 느낌!!

 

(품질경영기사도 품질이랑 경영이랑 같이 문제를 내다 보니 셤이 이도저도 아닌 느낌...)

 

(QC가 쓰는 부분은 통계 부분이 주였을텐데...)

 

 

 

그래도 일단은

 

필기를 붙었다는 가정 하에 실기를 공부해야 합니다📗

 

그래서 과거 빅분기 자료를 가지고 놀다가 문득,

 

'아 특정 값만 시각화를 다르게 하고 싶은데' 라는 생각이 들었고,

 

다른 분의 블로그를 참고해서 작성했습니다.

 

(링크 달아둘게요~)

 

 

 

그럼 본론으로 들어가시죠!

 

https://dyddl1993.tistory.com/38

[Matplotlib] max등 조건에 따라 색깔 다르게 표현하기

---

import matplotlib.pyplot as plt
import matplotlib

matplotlib.rcParams["font.family"] = "Malgun Gothic"

plt.figure(figsize=(12,8))

clrs = ["green" if (store_price.iloc[r][0] < store_price.max()[0]) else "red" for r in range(store_price.shape[0])]
plt.bar(x=store_price.index, height=store_price["총구매액"], align="center", data=store_price, color=clrs)

plt.xticks(fontsize=6)

plt.show()

 

plt.bar의 파라미터 중 색을 지정할 수 있는 'color=' 가 있습니다.

 

color를 'green'처럼 단일로 지정할 수 있지만,

 

['green', 'red'] 처럼 리스트로 지정하면 두 가지 색깔로 표현할 수 있습니다.

 

만약 표현하고자 하는 데이터 수가 두 개보다 많다면 'green', 'red', 'green', 'red' , 'green'...... 

 

이렇게 번갈아가면서 표현됩니다.

 

 

 

그렇다면 리스트 안에 데이터 수 만큼 색을 지정해주면 각기 다른 색으로 표현할 수 있다는 것이죠.

 

그래서 list comprehension으로 조건에 따라 색을 리스트에 저장하는 식을 사용했습니다.

 

clrs 변수에 있는 조건은, 데이터 프레임의 값이 최대값보다 작으면 'green'을 저장하게 했습니다.

 

그리고 값이 최대이면 'red'를 지정했습니다.

 

 

 

저 리스트를 그대로 'color=' 파라미터 안에 넣으면

 

아래와 같이

표현되는 것을 볼 수 있습니다.

 

(크리스마스 같아요 꼭ㅋㅋ🎄)

---

나중에 특정 값을 강조하고 싶을 때 사용하면 좋을 것이라는 생각이 들어

 

이렇게 글을 남깁니다~~~

 

그리고 누군가에게도 도움이 되기를💛

 

그럼 저는 다시 도망가 볼게요!!

 

뿅!!