Python plt.scatter詳解

一、plt.scatter簡介

plt.scatter是matplotlib庫中用於繪製散點圖的函數，能夠將兩組數據組合起來，形成橫縱坐標，用點的形式展現出來。其中，橫坐標為一組數據中的每個元素，縱坐標為另一組數據中的每個元素。

二、plt.scatter參數介紹

plt.scatter函數有以下幾個常用參數：

1. x：用於指定散點圖中橫坐標的數據。

2. y：用於指定散點圖中縱坐標的數據。

3. s：用於指定散點的大小。

4. c：用於指定散點的顏色。

5. alpha：用於指定散點的透明度。

6. marker：用於指定散點的形狀。

三、使用plt.scatter繪製散點圖

下面是一個簡單的使用plt.scatter繪製散點圖的代碼：

import matplotlib.pyplot as plt

# 假設有兩組數據
x = [1, 2, 3, 4, 5]
y = [2, 3, 1, 5, 4]

# 繪製散點圖
plt.scatter(x, y)

# 顯示圖像
plt.show()

代碼輸出的散點圖的橫坐標為1~5，縱坐標為2、3、1、5、4，如圖：

四、指定顏色和形狀

除了默認的散點圖顏色和形狀，plt.scatter還可以通過c和marker參數來指定散點的顏色和形狀。

下面是一個例子：

import matplotlib.pyplot as plt

# 假設有兩組數據
x = [1, 2, 3, 4, 5]
y = [2, 3, 1, 5, 4]

# 繪製散點圖，並指定顏色和形狀
plt.scatter(x, y, c=['r', 'g', 'b', 'y', 'k'], marker='o')

# 顯示圖像
plt.show()

代碼輸出的散點圖的橫坐標為1~5，縱坐標為2、3、1、5、4，每個點的顏色和形狀分別為[‘r’, ‘g’, ‘b’, ‘y’, ‘k’]和’o’，如圖：

五、指定大小和透明度

除了顏色和形狀，plt.scatter還可以通過s和alpha參數來指定散點的大小和透明度。

下面是一個例子：

import matplotlib.pyplot as plt

# 假設有兩組數據
x = [1, 2, 3, 4, 5]
y = [2, 3, 1, 5, 4]

# 繪製散點圖，並指定大小和透明度
plt.scatter(x, y, s=[20, 40, 60, 80, 100], alpha=0.5)

# 顯示圖像
plt.show()

代碼輸出的散點圖的橫坐標為1~5，縱坐標為2、3、1、5、4，每個點的大小分別為[20, 40, 60, 80, 100]，透明度為0.5，如圖：

六、使用顏色映射

使用顏色映射可以在散點圖中展示一個連續變化的信息，比如不同數據的值或概率分布情況。

下面是一個例子：

import matplotlib.pyplot as plt
import numpy as np

# 假設有兩組數據
x = np.random.rand(100)
y = np.random.rand(100)

# 隨機生成100個0~1之間的數
colors = np.random.rand(100)

# 繪製散點圖，並使用顏色映射
plt.scatter(x, y, c=colors, cmap='jet')

# 添加顏色映射條
plt.colorbar()

# 顯示圖像
plt.show()

代碼輸出的散點圖的橫坐標和縱坐標都是100個0~1之間的隨機數，每個散點的顏色根據0~1之間的隨機數生成，使用jet顏色映射。下圖展示了該散點圖的效果：

七、結語

到此為止，我們已經學習了使用plt.scatter繪製散點圖的基本方法以及常用參數。希望這篇文章能對你有所幫助。