Matplotlib散点图与NumPy数组：正确绘制单个坐标点

在使用Matplotlib的`scatter`函数结合NumPy数组绘制散点图时，若不正确地处理坐标输入，特别是当尝试从列向量中提取单个点的x和y坐标时，可能会意外地绘制出多个点。本文将深入解析这一常见误区，并通过具体示例演示如何正确地从NumPy数组中提取并传递单个点的x和y坐标给`scatter`函数，确保实现预期的单点绘制效果。

在数据可视化中，Matplotlib的pyplot.scatter()函数是绘制散点图的常用工具。它接受两个主要的参数：x坐标序列和y坐标序列。通常，当x和y是列表或NumPy数组时，它们被视为对应点的坐标集合，即scatter(x_values, y_values)会绘制一系列点(x_values[i], y_values[i])。然而，当输入数据结构为NumPy数组，特别是列向量时，如果不理解其索引机制，可能会导致非预期的绘图结果。

理解问题根源：NumPy数组的索引与scatter函数行为

我们首先通过几个示例来观察scatter函数的行为，特别是当输入数据源是NumPy数组时。

1. 使用Python列表绘制单个点（预期行为）

当使用简单的Python列表来存储坐标时，我们通常会通过索引直接获取x和y值。

import matplotlib.pyplot as plt
import numpy as np

# 示例1: 使用列表绘制单个点
a = [5, 6]
plt.scatter(a[0], a[1])
plt.title("Plotting a single point (5,6) with list")
plt.xlabel("X-axis")
plt.ylabel("Y-axis")
plt.grid(True)
plt.show()

上述代码会正确地在坐标(5,6)处绘制一个点。a[0]提供了x值5，a[1]提供了y值6。

2. 使用NumPy数组绘制单个点（x和y值相同）

当x和y坐标值相同时，即使使用NumPy数组的列向量形式，也可能因为巧合而得到预期结果。

# 示例2: 使用NumPy数组绘制单个点 (x=y)
b = np.array([[4], [4]])
plt.scatter(b[:, 0], b[:, 0])
plt.title("Plotting a single point (4,4) with NumPy array (x=y)")
plt.xlabel("X-axis")
plt.ylabel("Y-axis")
plt.grid(True)
plt.show()

这里，b[:, 0]会提取出NumPy数组[4, 4]。因此，plt.scatter接收到x_values=[4,4]和y_values=[4,4]，它会绘制点(4,4)。由于x和y值相同，最终只显示一个点，这可能让人误以为是正确处理了单个坐标。

OpenAI Codex

可以生成十多种编程语言的工作代码，基于 OpenAI GPT-3 的自然语言处理模型

144

查看详情

3. 使用NumPy数组绘制单个点（x和y值不同，导致问题）

当x和y坐标值不同，并且使用与示例2类似的方式进行索引时，问题就会浮现。

# 示例3: 使用NumPy数组绘制单个点 (x!=y, 错误示例)
c = np.array([[5], [6]])
print(f"c[:,0] 结果: {c[:,0]}") # 输出: [5 6]
plt.scatter(c[:, 0], c[:, 0])
plt.title("Problem: Plotting multiple points (5,5) and (6,6) instead of (5,6)")
plt.xlabel("X-axis")
plt.ylabel("Y-axis")
plt.grid(True)
plt.show()

在这个例子中，我们希望绘制单个点(5,6)。然而，c[:, 0]的结果是NumPy数组[5, 6]。因此，plt.scatter(c[:, 0], c[:, 0])实际上被解释为plt.scatter(x_values=[5, 6], y_values=[5, 6])。根据scatter函数的行为，它会绘制两个点：第一个x值与第一个y值配对形成(5,5)，第二个x值与第二个y值配对形成(6,6)。这显然不是我们想要的结果。

解决方案：正确提取单个点的x和y坐标

要解决上述问题，关键在于理解plt.scatter()在绘制单个点时，其x和y参数需要是标量值，而不是序列。当我们的坐标存储在一个NumPy数组（如列向量c）中时，我们需要直接索引到具体的标量元素。

正确的做法是，像处理Python列表a一样，通过索引获取NumPy数组中代表x和y的单个标量值。

import matplotlib.pyplot as plt
import numpy as np

# 示例4: 正确使用NumPy数组绘制单个点 (x!=y)
c = np.array([[5], [6]])

# 正确的做法：直接索引获取x和y的标量值
plt.scatter(c[0], c[1]) # c[0] 是 5, c[1] 是 6
plt.title("Solution: Correctly plotting a single point (5,6) with NumPy array")
plt.xlabel("X-axis")
plt.ylabel("Y-axis")
plt.grid(True)
plt.show()

在这个解决方案中：

• c[0]返回的是NumPy数组[5]，但当它作为plt.scatter的参数时，会被自动解包或视为标量值5。
• c[1]返回的是NumPy数组[6]，同样被视为标量值6。 因此，plt.scatter(c[0], c[1])实际上等同于plt.scatter(5, 6)，从而正确地绘制出单个点(5,6)。

注意事项与总结

• plt.scatter(x, y)参数类型：
  • 当绘制单个点时，x和y应为标量值（例如，plt.scatter(5, 6)）。
  • 当绘制多个点时，x和y应为等长的序列（例如，plt.scatter([1,2,3], [4,5,6])）。
• NumPy数组索引：
  • array[:, 0]：这会从所有行中提取第0列，结果是一个一维NumPy数组。如果原数组是[[x],[y]]，则array[:,0]会得到[x, y]。
  • array[0]：这会提取第0行。如果原数组是[[x],[y]]，则array[0]会得到[x]。
  • array[0, 0]：这会提取第0行第0列的标量值。如果原数组是[[x],[y]]，则array[0,0]会得到x。
• 推荐做法： 当NumPy数组存储的是单个点的坐标时，为了清晰和避免混淆，建议直接通过索引获取标量值，例如 plt.scatter(c[0, 0], c[1, 0])，或者如果数组结构允许，使用 c[0] 和 c[1] 这种方式，它在大多数情况下也会被正确解释为标量。最重要的是确保传递给scatter函数的是单个的x和y坐标值，而不是包含多个x或y值的序列。

通过理解scatter函数的参数期望以及NumPy数组的索引行为，我们可以避免在绘制散点图时出现意外的多点现象，确保数据可视化的准确性。

以上就是Matplotlib散点图与NumPy数组：正确绘制单个坐标点的详细内容，更多请关注php中文网其它相关文章！