在Matplotlib中,可以使用scatter
方法來繪制散點圖,并通過傳入參數c
或s
來將數據點按顏色或大小映射到其他變量。以下是一個示例代碼:
import matplotlib.pyplot as plt
import numpy as np
# 生成隨機數據
np.random.seed(0)
x = np.random.rand(100)
y = np.random.rand(100)
colors = np.random.rand(100)
sizes = 1000 * np.random.rand(100)
plt.scatter(x, y, c=colors, s=sizes, alpha=0.5, cmap='viridis')
plt.colorbar() # 添加顏色條
plt.show()
在上面的示例代碼中,c=colors
將數據點按顏色映射到colors
變量的值,s=sizes
將數據點按大小映射到sizes
變量的值。可以通過調整sizes
和colors
的值來改變數據點的大小和顏色。alpha
參數用于設置數據點的透明度,cmap
參數用于設置顏色映射的顏色映射方案。最后,使用colorbar
方法添加顏色條,顯示顏色映射的范圍。