时间戳

首页 > 解决方案 > 在python中的矩阵中的相同值之间绘制网格线

python - 在python中的矩阵中的相同值之间绘制网格线

问题描述

我在python中有一个np.array，如下所示：

mat = np.array([[0.2, 0.2, 0.1, 0.1, 0.1],
      [0.2, 0.2, 0.1, 0.1, 0.0],
      [0.2, 0.2, 0.1, 0.1, 0.0],
      [0.2, 0.1, 0.1, 0.0, 0.0]])
mat

输出

我想在这个数组中的相同值之间画线，这样：

带网格线的输出

最后我想有一个这样的情节：

网格图

我在 python 中寻找 matplotlib 和 turtle 库，但找不到绘制它的方法。

标签： pythonmatplotlibturtle-graphicspython-turtle

解决方案

仅使用 matplotlib ：

import matplotlib.pyplot as plt

mat = [[0.2, 0.2, 0.1, 0.1, 0.1],
      [0.2, 0.2, 0.1, 0.1, 0.0],
      [0.2, 0.2, 0.1, 0.1, 0.0],
      [0.2, 0.1, 0.1, 0.0, 0.0]]

def line(x,y):
     plt.plot(x,y,marker = 'o',color = 'red',
     markerfacecolor='black',markeredgecolor='black')

def draw(mat):
     for i in range(len(mat)):
          for k in range(len(mat[i])-1):
               if mat[i][k] == mat[i][k+1] :
                    line([k,k+1],[len(mat)-i,len(mat)-i])
     for i in range(len(mat)-1):
          for k in range(len(mat[i])):
               if mat[i][k] == mat[i+1][k]:
                    line([k,k],[len(mat)-i,len(mat)-i-1])

draw(mat)
plt.show()

推荐阅读