Python实现矩阵可视化的示例代码

 更新时间:2023年10月19日 09:53:22   作者:微小冷  
matplotlib中提供了两个矩阵可视化函数,分别是imshow和matshow,本文主要为大家详细介绍了如何使用这两个函数,感兴趣的小伙伴可以跟随小编一起学习一下

imshow和matshow

matplotlib中提供了两个矩阵可视化函数,分别是imshow和matshow,由于二者过于相似,且imshow常被用做图片展示工具,所以matshoww这个函数基本没什么人知道,总之二者对比如下

import matplotlib.pyplot as plt
import numpy as np

x = np.random.rand(5,5)

fig,axes = plt.subplots(1,2,figsize=(8,4))

axes[0].imshow(x)
plt.title("imshow")
axes[1].matshow(x)
plt.title("matshow")

plt.tight_layout()
plt.show()

但无论是imshow也好,还是matshow也罢,对矩阵,尤其是小矩阵,展示得并不完美,徒有伪彩色对应,而并无具体的数值,那么对于一些需要看到实际数值的场合,其表现力当然是不够的。

显示数值

所以接下来,就要在特定的格子中,写下矩阵的具体数值。

from itertools import product

M,N = 3,6
x = np.random.rand(M,N)

plt.matshow(x)
for i,j in product(range(M),range(N)):
    plt.text(j-0.15, i, f"{x[i,j]:.2}")

plt.show()

product是排列组合迭代器,可以将输入序列的元素两两组合在一起,从而避免循环嵌套。plt.text的作用就是显示数值,其中j表示x坐标,i表示y轴坐标。在矩阵索引中,i表示行号,j表示列号。

这张图当然也有问题,由于文字是横着写的,但矩阵中的格子却是正方形的,这个显然看上去不太和谐。如果用pcolormesh这种像素形状可以更改的函数,显然更加合适。

但pcolormesh有两个问题,一是坐标轴方向和矩阵下标的方向不符,二是坐标标签的位置并不指格点中间,为此需要稍加改造。

def drawMat(x, ax=None):
    M, N = x.shape
    if not ax:
        ax = plt.subplot()
    arrM, arrN = np.arange(M), np.arange(N)
    plt.yticks(arrM+0.5, arrM)
    plt.xticks(arrN+0.5, arrN)
    ax.pcolormesh(x)
    ax.invert_yaxis()
    for i,j in product(range(M),range(N)):
        ax.text(j+0.2, i+0.55, f"{x[i,j]:.2}")
    plt.show()

x = np.random.rand(5,5)
drawMat(x)

其中,xticks和yticks用于重新映射坐标,将N.5映射为N,这样坐标位置也就转化为了具体数值;invert_yaxis表示y轴坐标翻转,从而直角坐标系被改为矩阵坐标系。

效果如下

到此这篇关于Python实现矩阵可视化的示例代码的文章就介绍到这了,更多相关python矩阵可视化内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

相关文章

  • 从基础操作到高效实践详解DataFrame数据修改的完整指南

    从基础操作到高效实践详解DataFrame数据修改的完整指南

    数据修改(Data Mutation)是解决这些问题的核心能力,它直接影响后续分析的准确性,本文将通过 6类操作 + 4个实战案例,系统讲解如何安全、高效地修改DataFrame数据,感兴趣的小伙伴可以了解下
    2026-02-02
  • windows 下python+numpy安装实用教程

    windows 下python+numpy安装实用教程

    这篇文章主要介绍了windows 下python+numpy安装实用教程,具有一定借鉴价值,需要的朋友可以参考下。
    2017-12-12
  • python爬虫lxml库解析xpath网页过程示例

    python爬虫lxml库解析xpath网页过程示例

    这篇文章主要为大家介绍了python爬虫lxml库解析xpath网页的过程示例,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪
    2022-05-05
  • Python 高效编程技巧分享

    Python 高效编程技巧分享

    工作中经常要处理各种各样的数据,遇到项目赶进度的时候自己写函数容易浪费时间。Python 中有很多内置函数帮你提高工作效率。
    2020-09-09
  • pycharm中使用pyplot时报错MatplotlibDeprecationWarning

    pycharm中使用pyplot时报错MatplotlibDeprecationWarning

    最近在使用Pycharm中matplotlib作图处理时报错,所以这篇文章主要给大家介绍了关于pycharm中使用pyplot时报错MatplotlibDeprecationWarning的相关资料,需要的朋友可以参考下
    2023-12-12
  • Python使用py2exe打包程序介绍

    Python使用py2exe打包程序介绍

    这篇文章主要介绍了Python使用py2exe打包程序介绍,本文讲解了py2exe简介、安装、用法、指定额外文件等内容,需要的朋友可以参考下
    2014-11-11
  • python打开网页和暂停实例

    python打开网页和暂停实例

    这篇文章主要介绍了python打开网页和暂停实例,涉及webbrowser模块与os模块的使用,是非常具有实用价值的一个技巧,需要的朋友可以参考下
    2014-09-09
  • Python中如何创建和运行异步任务详解

    Python中如何创建和运行异步任务详解

    这篇文章主要为大家介绍了Python中如何创建和运行异步任务详解,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪
    2023-03-03
  • python基于paramiko将文件上传到服务器代码实现

    python基于paramiko将文件上传到服务器代码实现

    这篇文章主要介绍了python基于paramiko将文件上传到服务器代码实现,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
    2019-07-07
  • 关于查找numpy和pandas兼容的版本要求

    关于查找numpy和pandas兼容的版本要求

    本文主要介绍了关于查找numpy和pandas兼容的版本要求,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2026-01-01

最新评论