赞
踩
本次演示用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:
本人学习过程中的一些笔记,文中可能有些不足,欢迎大家讨论和指正。
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。