Computer >> 컴퓨터 >  >> 프로그램 작성 >> Python

조건이 true인 경우 단일 막대의 색상을 변경하는 방법(Matplotlib)?

<시간/>

조건이 true인 경우 단일 막대의 색상을 변경하려면 값 세트와 값이 2가 될 때까지 빨간색으로 색상 목록을 만들 수 있습니다. 그렇지 않으면 목록에 노란색을 추가합니다.

단계

  • 그림 크기를 설정하고 서브플롯 사이 및 주변 여백을 조정합니다.
  • 바의 가변 너비를 초기화합니다.
  • 값과 색상 목록을 두 개 만드세요.
  • bar() 사용 막대를 그리는 방법.
  • 그림을 표시하려면 show()를 사용하세요. 방법.

예시

import numpy as np
import matplotlib.pyplot as plt
plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True
data = np.arange(5)
width = 0.5
vals = [1, 2, 1, 5, 3]
colors = ["red" if i != 2 else "yellow" for i in vals]
plt.bar(data, vals, width, color=colors)
plt.show()

출력

조건이 true인 경우 단일 막대의 색상을 변경하는 방법(Matplotlib)?