如何在散點圖中為 Matplotlib 建立分離色條?


使用 plt.colorbar(ticks=np.linspace(-2, 2, 5)),我們可以建立一個離散色條。

步驟

  • 使用 np.random.random 方法返回半開區間中的隨機浮點數,即 x。

  • 使用 np.random.random 方法返回半開區間中的隨機浮點數,即 y。

  • 使用 np.random.randint(-2, 3, 20) 方法返回從“低”值(包含)到“高”值(不含)的隨機整數,即 z。

  • 使用 plt.xlabel() 設定 X 軸標籤。

  • 使用 plt.ylabel() 設定 Y 軸標籤。

  • 使用內建彩虹色圖。

  • 根據離散間隔生成色圖索引。

  • 建立一個包含可變標記大小和/或顏色的 *y* 對 *x* 的散點圖,其中 x、y 和 z 已建立(步驟 1、2、3)。

  • 為 ScalarMappable 例項 *mappable* 建立色條。

  • 使用 plt.show() 來顯示圖形。

示例

from matplotlib import pyplot as plt
import numpy as np
from matplotlib import colors

x = np.random.random(20)
y = np.random.random(20)
z = np.random.randint(-2, 3, 20)

plt.xlabel('X-axis ')
plt.ylabel('Y-axis ')

cmap = plt.cm.rainbow
norm = colors.BoundaryNorm(np.arange(-2.5, 3, 1), cmap.N)

plt.scatter(x, y, c=z, cmap=cmap, norm=norm, s=100, edgecolor='none')
plt.colorbar(ticks=np.linspace(-2, 2, 5))

plt.show()

輸出

更新於:2021-03-15

7 千次瀏覽

啟動你的 事業

完成課程,獲得認證

開始
廣告
© . All rights reserved.