使用Matplotlib绘制散点图：理解NumPy数组输入与坐标点映射

在使用`matplotlib.pyplot.scatter`绘制散点图时，若输入为NumPy数组，需特别注意其维度和索引方式，以确保正确映射单个坐标点或多个数据点。本文将详细解析`scatter`函数处理NumPy数组的机制，并提供正确的输入方法，避免因误解数组切片而导致绘制出非预期的多个点。

理解matplotlib.pyplot.scatter函数的工作原理

matplotlib.pyplot.scatter函数用于绘制二维散点图，它接受两个主要的参数：x和y，分别代表所有点的x坐标序列和y坐标序列。其核心机制是将x序列的第i个元素与y序列的第i个元素配对，形成一个坐标点(x[i], y[i])。当x和y是单个数值（或包含单个元素的序列）时，它会绘制一个点；当它们是包含多个元素的序列时，它会绘制对应数量的散点。

列表输入：直观的单点绘制

对于Python列表，绘制单个点通常非常直观。例如，要绘制点(5, 6)：

import matplotlib.pyplot as plt
import numpy as np

a = [5, 6]
plt.scatter(a[0], a[1]) # x=5, y=6
plt.title("Single Point from List")
plt.show()

登录后复制

此代码会正确绘制一个位于(5, 6)的散点，因为a[0]提供了x坐标，a[1]提供了y坐标。

NumPy数组输入：单点与多点的区分

当使用NumPy数组作为输入时，需要特别注意数组的结构和切片操作。

示例1：NumPy数组（同值）的单点绘制

如果NumPy数组的结构使得x和y输入都只包含一个元素，即使是数组形式，也会被正确解析为单点。

b = np.array([[4],[4]]) # b 是一个 2x1 的数组
plt.scatter(b[:,0], b[:,0]) # x=[4], y=[4]
plt.title("Single Point from NumPy (Same Values)")
plt.show()

登录后复制

在此例中，b[:,0]的结果是array([4, 4])。然而，当plt.scatter接收到 x = np.array([4]) 和 y = np.array([4]) 时，它会绘制一个点(4, 4)。实际上，原始代码 plt.scatter(b[:,0], b[:,0]) 会将 x 和 y 都设为 array([4, 4])，因此会绘制两个点 (4, 4) 和 (4, 4)，它们重叠在一起，看起来像一个点。为了更清晰地表示单点，我们应该确保x和y输入确实是单个值。

标签： python