如何将x、y数据存为列后在matplotlib绘图并通过列访问子图

核心调整思路 #

• 以Python内置字典作为统一列存储结构，按“绘图标识-关联内容”的键值对组织x、y数据与对应的子图对象，支持通过标识键直接索引访问，完全匹配“通过列结构直接访问对应子图”的需求
• 切换为Matplotlib面向对象绘图接口，替代零散的全局变量定义与pyplot状态机调用，将轴对象和绑定的绘图数据存入同一存储结构，避免全局变量混乱
• 完全保留原有蓝色三角形的绘制逻辑，不改动原始坐标数据的视觉效果

调整后的完整可运行代码 #

import numpy as np
import matplotlib.pyplot as plt

# 定义统一列存储结构：键为绘图项目的唯一标识，值为对应的数据、关联子图对象
plot_column_store = {
    "blue_triangle": {
        # 原样存入原始x、y列坐标数据
        "x_data": np.array([0, 0, 4, 0]),
        "y_data": np.array([0, 3, 3, 0]),
        # 预留字段存储关联的子图对象，完成数据与子图的绑定
        "axes": None
    }
}

# 创建画布与初始绘图子图
fig = plt.figure()
ax = fig.add_subplot(111)
# 将生成的子图对象存入列结构，建立绑定关系
plot_column_store["blue_triangle"]["axes"] = ax

# 直接从列结构读取x、y数据完成绘制
target_entry = plot_column_store["blue_triangle"]
target_entry["axes"].plot(
    target_entry["x_data"],
    target_entry["y_data"],
    color="blue"
)

plt.show()

列结构访问方式 #

后续所有对该绘图项的操作都可以通过列结构的键索引直接完成，无需查找零散的全局变量：

• 读取对应x坐标数据：plot_column_store["blue_triangle"]["x_data"]
• 读取对应y坐标数据：plot_column_store["blue_triangle"]["y_data"]
• 获取关联的子图对象，直接在子图上做后续修改：

  # 直接通过列结构拿到绑定的子图
  triangle_ax = plot_column_store["blue_triangle"]["axes"]
  # 给子图加标题、调整纵横比的示例操作
  triangle_ax.set_title("直角三角形")
  triangle_ax.set_aspect("equal")
  

如果后续需要新增多张子图，只需要在plot_column_store中追加新的键值对条目，存入对应x、y数据和关联的子图对象即可，访问逻辑保持统一。

内容的提问来源于stack exchange，提问作者brian.p