溫馨提示×

  • 首頁 > 
  • 問答 > 
  • 編程語言  > 
  • Matplotlib怎么將數(shù)據(jù)點(diǎn)按大小或顏色映射到其他變量

Matplotlib怎么將數(shù)據(jù)點(diǎn)按大小或顏色映射到其他變量

小億
83
2024-05-21 11:29:29
欄目: 編程語言

在Matplotlib中,可以使用scatter方法來繪制散點(diǎn)圖,并通過傳入?yún)?shù)cs來將數(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)整sizescolors的值來改變數(shù)據(jù)點(diǎn)的大小和顏色。alpha參數(shù)用于設(shè)置數(shù)據(jù)點(diǎn)的透明度,cmap參數(shù)用于設(shè)置顏色映射的顏色映射方案。最后,使用colorbar方法添加顏色條,顯示顏色映射的范圍。

0