要在Matplotlib中繪制散點圖以顯示類別變量之間的關系,可以使用不同顏色或標記符號來區分不同的類別。可以通過以下步驟實現:
以下是一個示例代碼,演示如何繪制一個包含類別變量的散點圖:
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()
運行以上代碼將會得到一個顯示類別變量之間關系的散點圖,其中不同類別用不同顏色或標記符號表示。您可以根據實際數據集的情況來自定義繪圖風格和參數。