使用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输入确实是单个值。

捏Ta

捏Ta 是一个专注于角色故事智能创作的AI漫画生成平台

322

查看详情

示例2：NumPy数组（不同值）导致的常见误解

这是最容易出现问题的情况。假设我们有一个NumPy数组c，我们希望从中提取x和y坐标来绘制一个点(5, 6)。

c = np.array([[5],[6]]) # c 是一个 2x1 的数组
print(f"c[:,0] 的结果是: {c[:,0]}") # 输出: c[:,0] 的结果是: [5 6]

plt.scatter(c[:,0], c[:,0]) # x=[5,6], y=[5,6]
plt.title("Problematic Plot: Unexpected Multiple Points")
plt.show()

问题分析： 在此示例中，c是一个2行1列的NumPy数组。c[:,0]的切片操作会提取所有行的第0列元素，结果是array([5, 6])。因此，plt.scatter(c[:,0], c[:,0])实际上等价于plt.scatter(np.array([5, 6]), np.array([5, 6]))。

根据scatter函数的工作原理，它会将第一个序列的第一个元素与第二个序列的第一个元素配对，形成第一个点(5, 5)；然后将第一个序列的第二个元素与第二个序列的第二个元素配对，形成第二个点(6, 6)。最终，图上会显示两个点：(5, 5)和(6, 6)，而不是预期的单个点(5, 6)。

正确绘制单个点的解决方案

要从NumPy数组c = np.array([[5],[6]])中提取出x坐标5和y坐标6来绘制单个点(5, 6)，我们需要像处理列表一样，分别访问数组的对应元素。

c = np.array([[5],[6]])

# 正确的方法：分别提取x和y坐标
plt.scatter(c[0], c[1]) # x=c[0] (即 [5]), y=c[1] (即 [6])
plt.title("Correct Single Point from NumPy")
plt.show()

解决方案解释： 当c = np.array([[5],[6]])时：

• c[0]会返回数组的第一行，即np.array([5])。
• c[1]会返回数组的第二行，即np.array([6])。

plt.scatter函数能够识别并正确处理这些包含单个元素的NumPy数组，将其解释为单个x坐标5和单个y坐标6，从而绘制出期望的单个点(5, 6)。

总结与最佳实践

在使用matplotlib.pyplot.scatter与NumPy数组时，请牢记以下几点：

1. 理解scatter的输入： scatter(x_sequence, y_sequence)期望两个序列，其中x_sequence[i]和y_sequence[i]构成第i个点。
2. 区分单点与多点：
   • 绘制单个点(x_val, y_val)： 确保scatter的第一个参数解析为x_val，第二个参数解析为y_val。这可以通过直接传递标量、列表的单个元素或NumPy数组的单个元素（如arr[0]、arr[1]）来实现。
   • 绘制多个点： 确保scatter的第一个参数是一个包含所有x坐标的序列，第二个参数是一个包含所有y坐标的序列，且两者长度相等。
3. NumPy数组切片： array[:,0]会提取指定列的所有元素，形成一个新的序列。如果您的数组是列向量形式（例如[[x1],[y1]]），并且您想绘制单个点(x1, y1)，则不应使用array[:,0]作为两个输入，而应该分别访问array[0]和array[1]。

通过清晰地理解NumPy数组的结构和matplotlib.pyplot.scatter的输入要求，您可以避免常见的陷阱，并准确地绘制出所需的散点图。

以上就是使用Matplotlib绘制散点图：理解NumPy数组输入与坐标点映射的详细内容，更多请关注php中文网其它相关文章！