0

0

如何用循环简化 Matplotlib 中的四视图 3D 散点图绘制

霞舞

霞舞

发布时间:2025-12-27 18:44:02

|

556人浏览过

|

来源于php中文网

原创

如何用循环简化 Matplotlib 中的四视图 3D 散点图绘制

本文介绍如何通过 for 循环重构重复性高的 `scatter3d` 多子图代码,统一设置坐标轴、颜色映射与视角参数,显著提升可读性与可维护性。

在使用 Matplotlib 绘制多视角 3D 散点图时,若为每个子图单独编写几乎相同的绘图逻辑(仅视角 view_init(elev, azim) 不同),不仅代码冗长,还容易因漏改某处参数导致不一致。最佳实践是将共性逻辑提取为循环体,差异化部分(如视角角度)以元组列表或函数生成方式传入。

以下是一个优化后的专业写法,支持任意数量的子图,并清晰分离「数据绘制」「视角配置」和「坐标系设置」三类操作:

import matplotlib.pyplot as plt

def plot_example(data_df):
    """
    绘制四视角 3D 散点图:俯视(0°,90°)、侧视(45°,0°)、斜视1(35°,45°)、斜视2(20°,40°)

    Parameters:
    -----------
    data_df : pandas.DataFrame
        必须包含 'x', 'y', 'z' 列
    """
    fig = plt.figure(figsize=(15, 15))

    # 预定义各子图视角:(elev, azim)
    view_angles = [(0, 90), (45, 0), (35, 45), (20, 40)]

    for i, (elev, azim) in enumerate(view_angles, start=1):
        ax = fig.add_subplot(2, 2, i, projection='3d')

        # 统一绘图
        scatter = ax.scatter3D(
            data_df.x, data_df.y, data_df.z,
            c=data_df.z, cmap='Blues', s=20, alpha=0.8
        )

        # 设置视角
        ax.view_init(elev=elev, azim=azim)

        # 统一标签与样式
        ax.set_xlabel('x', color='red', fontsize=10)
        ax.set_ylabel('y', color='red', fontsize=10)
        ax.set_zlabel('z', color='red', fontsize=10)

        # 统一坐标范围(关键:必须对每个 ax 单独调用)
        ax.set_xlim(0, 14)
        ax.set_ylim(-6, 6)
        ax.set_zlim(0, 8.5)

    # 可选:添加共享 colorbar(增强可读性)
    cbar = fig.colorbar(scatter, ax=fig.axes, shrink=0.5, aspect=20, pad=0.1)
    cbar.set_label('z value', color='red', fontsize=12)

    plt.tight_layout()
    plt.show()

关键优化点说明

Mureka
Mureka

Mureka是昆仑万维最新推出的一款AI音乐创作工具,输入歌词即可生成完整专属歌曲。

下载
  • 避免 plt.axis() 误用:该函数仅适用于 2D 坐标系(接受 [xmin,xmax,ymin,ymax]),3D 场景中必须对每个 Axes3D 实例分别调用 set_xlim()/set_ylim()/set_zlim();
  • 视角解耦:将 (elev, azim) 显式定义为元组列表,语义清晰、便于调试与扩展;
  • 增强鲁棒性:添加 s(点大小)和 alpha(透明度)提升视觉层次;tight_layout() 自动优化子图间距;
  • 可扩展设计:如需增加第五个视角,只需在 view_angles 中追加元组,无需新增代码块。

⚠️ 注意事项

  • 确保传入的 data_df 已正确加载且无缺失值(NaN 会导致 scatter3D 报错);
  • 若数据量极大(>10⁵ 点),建议启用 antialiased=False 或降采样以提升渲染性能;
  • projection='3d' 仅支持 add_subplot() 创建,不兼容 plt.subplots() 的简写形式(需显式指定 subplot_kw={'projection': '3d'})。

通过结构化循环与参数化配置,原始 60+ 行重复代码可压缩至 30 行以内,同时大幅提升逻辑一致性与后期维护效率。

相关专题

更多
ip地址修改教程大全
ip地址修改教程大全

本专题整合了ip地址修改教程大全,阅读下面的文章自行寻找合适的解决教程。

33

2025.12.26

压缩文件加密教程汇总
压缩文件加密教程汇总

本专题整合了压缩文件加密教程,阅读专题下面的文章了解更多详细教程。

18

2025.12.26

wifi无ip分配
wifi无ip分配

本专题整合了wifi无ip分配相关教程,阅读专题下面的文章了解更多详细教程。

46

2025.12.26

漫蛙漫画入口网址
漫蛙漫画入口网址

本专题整合了漫蛙入口网址大全,阅读下面的文章领取更多入口。

91

2025.12.26

b站看视频入口合集
b站看视频入口合集

本专题整合了b站哔哩哔哩相关入口合集,阅读下面的文章查看更多入口。

283

2025.12.26

俄罗斯搜索引擎yandex入口汇总
俄罗斯搜索引擎yandex入口汇总

本专题整合了俄罗斯搜索引擎yandex相关入口合集,阅读下面的文章查看更多入口。

370

2025.12.26

虚拟号码教程汇总
虚拟号码教程汇总

本专题整合了虚拟号码接收验证码相关教程,阅读下面的文章了解更多详细操作。

35

2025.12.25

错误代码dns_probe_possible
错误代码dns_probe_possible

本专题整合了电脑无法打开网页显示错误代码dns_probe_possible解决方法,阅读专题下面的文章了解更多处理方案。

25

2025.12.25

网页undefined啥意思
网页undefined啥意思

本专题整合了undefined相关内容,阅读下面的文章了解更多详细内容。后续继续更新。

72

2025.12.25

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
誉天教育RHCE视频教程
誉天教育RHCE视频教程

共9课时 | 1.4万人学习

尚观Linux RHCE视频教程(二)
尚观Linux RHCE视频教程(二)

共34课时 | 5.6万人学习

尚观RHCE视频教程(一)
尚观RHCE视频教程(一)

共28课时 | 4.7万人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2025 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号