在Matplotlib中,可以使用scatter
方法來繪制散點(diǎn)圖,并通過傳入?yún)?shù)c
或s
來將數(shù)據(jù)點(diǎn)按顏色或大小映射到其他變量。以下是一個示例代碼:
import matplotlib.pyplot as plt
import numpy as np
# 生成隨機(jī)數(shù)據(jù)
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
將數(shù)據(jù)點(diǎn)按顏色映射到colors
變量的值,s=sizes
將數(shù)據(jù)點(diǎn)按大小映射到sizes
變量的值??梢酝ㄟ^調(diào)整sizes
和colors
的值來改變數(shù)據(jù)點(diǎn)的大小和顏色。alpha
參數(shù)用于設(shè)置數(shù)據(jù)點(diǎn)的透明度,cmap
參數(shù)用于設(shè)置顏色映射的顏色映射方案。最后,使用colorbar
方法添加顏色條,顯示顏色映射的范圍。