溫馨提示×

Matplotlib中怎么繪制散點圖以顯示類別變量之間的關系

小億
117
2024-05-20 19:39:34
欄目: 編程語言

要在Matplotlib中繪制散點圖以顯示類別變量之間的關系,可以使用不同顏色或標記符號來區分不同的類別??梢酝ㄟ^以下步驟實現:

  1. 創建一個包含類別變量的數據集,每個數據點包含兩個數值變量和一個類別變量。
  2. 使用不同顏色或標記符號來表示不同類別的數據點。
  3. 使用Matplotlib的scatter函數繪制散點圖,并傳入不同類別的數據點及其對應的顏色或標記符號。
  4. 添加圖例來說明不同類別對應的顏色或標記符號。

以下是一個示例代碼,演示如何繪制一個包含類別變量的散點圖:

import matplotlib.pyplot as plt
import numpy as np

# 生成隨機數據集
np.random.seed(0)
n = 100
x = np.random.rand(n)
y = np.random.rand(n)
category = np.random.choice(['A', 'B', 'C'], n)

# 繪制散點圖
plt.figure(figsize=(8, 6))
for cat in np.unique(category):
    plt.scatter(x[category == cat], y[category == cat], label=cat)

plt.xlabel('X')
plt.ylabel('Y')
plt.title('Scatter Plot with Categorical Variable')
plt.legend()
plt.show()

運行以上代碼將會得到一個顯示類別變量之間關系的散點圖,其中不同類別用不同顏色或標記符號表示。您可以根據實際數據集的情況來自定義繪圖風格和參數。

0
亚洲午夜精品一区二区_中文无码日韩欧免_久久香蕉精品视频_欧美主播一区二区三区美女