Matplotlib 산점도에 대한 사용자 정의 이산 색상 막대를 만드는 방법은 무엇입니까?

Matplotlib Scatterplot용 사용자 정의 개별 색상 막대

Matplotlib에서 산점도용 개별 색상 막대를 생성하면 데이터 포인트를 고유한 색상 기반으로 표현할 수 있습니다. 정수 태그 값에 대해. 이를 달성하기 위해 사용자 정의 이산 색상맵과 BoundaryNorm을 사용할 수 있습니다.

임의의 x 및 y 데이터로 산점도를 구성하고 0에서 20 사이의 정수 태그를 할당해 보겠습니다.

<code class="python">import matplotlib.pyplot as plt
import numpy as np

x = np.random.rand(20)
y = np.random.rand(20)
tag = np.random.randint(0, 20, 20)</code>

처음에는 기본 설정을 사용하여 컬러바에 연속적인 색상 범위가 표시됩니다.

<code class="python">plt.scatter(x, y, c=tag)
plt.colorbar()</code>

이산 컬러바를 생성하기 위해 BoundaryNorm을 사용하여 컬러맵의 경계를 정의합니다. 또한 태그 값 0이 회색으로 표시되는지 확인하려고 합니다.

<code class="python">bounds = np.linspace(0, 20, 21)
norm = mpl.colors.BoundaryNorm(bounds, cmap.N)

cmaplist = [cmap(i) for i in range(cmap.N)]
cmaplist[0] = (.5, .5, .5, 1.0)

cmap = mpl.colors.LinearSegmentedColormap.from_list(
    'Custom cmap', cmaplist, cmap.N)</code>

이산 컬러바가 포함된 업데이트된 산점도는 다음과 같습니다.

<code class="python">scat = ax.scatter(x, y, c=tag, s=np.random.randint(100, 500, 20),
                  cmap=cmap, norm=norm)

# Add the discrete colorbar
ax2 = fig.add_axes([0.95, 0.1, 0.03, 0.8])
cb = plt.colorbar.ColorbarBase(ax2, cmap=cmap, norm=norm,
    spacing='proportional', ticks=bounds, boundaries=bounds, format='%1i')</code>

이 접근 방식은 다음과 같습니다. 각 태그 값은 태그 값 0의 회색을 포함하여 고유한 색상으로 표시되는 분산형 차트에 대해 잘 정의된 개별 색상 막대입니다.

위 내용은 Matplotlib 산점도에 대한 사용자 정의 이산 색상 막대를 만드는 방법은 무엇입니까?의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!