python用matplotlib 用matshow（）绘制矩阵，绘制矩阵图

2023-12-13 17:48| 来源: 网络整理| 查看: 265

本次演示用jupter notebook.

一、plt.cm绘制

示例一：设置0-100的10*10的矩阵

代码1：

import matplotlib.pyplot as plt import numpy as np A = np.arange(0, 100).reshape(10, 10) plt.matshow(mat, cmap=plt.cm.Reds)#这里设置颜色为红色，也可以设置其他颜色 plt.title("matrix A") plt.show()

得到如下结果图1：

注：对于矩阵的颜色设置还有很多，改变 plt.matshow(mat, cmap=plt.cm.Reds)中的Reds可根据自己喜好设置其他颜色，以下给出其他几种颜色。

代码2：（设置颜色为蓝色）

import matplotlib.pyplot as plt import numpy as np A = np.arange(0, 100).reshape(10, 10) plt.matshow(mat, cmap=plt.cm.Blues) plt.title("matrix A") plt.show()

结果2：

代码3：（设置颜色灰色）

import matplotlib.pyplot as plt import numpy as np A = np.arange(0, 100).reshape(10, 10) plt.matshow(mat, cmap=plt.cm.gray) plt.title("matrix A") plt.show()

结果3：

示例二：随机设置矩阵 A

代码4：

import matplotlib.pyplot as plt import numpy as np A=np.array([[4,3,2,4],[5,4,7,8],[9,16,11,5],[13,3,4,16],[6,18,1,20]]) plt.matshow(X, cmap=plt.cm.Reds) plt.title("matrix A") plt.show()

结果4：

注：从图中可以看出，不同的数值大小红色的深度是不一样的，数值越大红色越深，这里同样可以根据自己喜好设置不同的颜色，我就不再过多展示。

二、用plt.colorbar绘制矩阵图

代码5：

import numpy as np import matplotlib.pyplot as plt X=np.array([[0,3,2,4],[5,4,7,8],[9,16,8,5],[13,3,4,16],[6,18,1,20]]) A = np.arange(0, 100).reshape(10, 10) ax = plt.matshow(X) plt.colorbar(ax.colorbar, fraction=0.025) plt.title("matrix X"); plt.show()

结果5：

本人学习过程中的一些笔记，文中可能有些不足，欢迎大家讨论和指正。

【本文地址】

公司简介

联系我们