用matplotlib制作代码和色块

代码如下：

# 声明
# -*- coding: utf-8 -*-
"""
Created on Mon May 13 11:18:59 2024

@author: sankang
"""
# 这里调用包
import matplotlib as mpl
import matplotlib.pyplot as plt
import numpy as np

plt.rcParams['axes.unicode_minus']=False
n=3
x=np.linspace(-10,10,5)
y=np.linspace(-10,10,n)
# 构造网格点
X,Y=np.meshgrid(x,y)
# 使用散点图绘制网格点
plt.scatter(X,Y,s=60,c='r',marker='x')
plt.show()
# 设置类别号
Z=np.array([[1,2,3,4],[2,1,4,3]])
# 这里注释的是用来测试数据是否传入
# print(Z.shape)
# 作色块图
cm_light=mpl.colors.ListedColormap(['y','r','g','b'])
# pcolormesh()中，X,Y是坐标值，Z是类别值，camp是颜色值
plt.pcolormesh(X,Y,Z,cmap=cm_light,alpha=0.5)
# 输出
plt.show()

效果如下：