Um gráfico de matriz de grade 2D pode ser uma ferramenta de visualização valiosa, por exemplo, na área de simulação baseada em agentes. Neste post eu quero dar um breve tutorial de como você pode visualizar um grid array 2D, usando matplotlib em Python. O exemplo de codificação está abaixo; documentação relevante foi adicionada na forma de comentários.

from matplotlib import pyplot
import random
data = [[random.randint(a=0,b=1) for x in range(0,8)], # row 1
        [random.randint(a=0,b=1) for x in range(0,8)], # row 2
        [random.randint(a=0,b=1) for x in range(0,8)], # row 3
        [random.randint(a=0,b=1) for x in range(0,8)], # row 4
        [random.randint(a=0,b=1) for x in range(0,8)], # row 5
        [random.randint(a=0,b=1) for x in range(0,8)], # row 6
        [random.randint(a=0,b=1) for x in range(0,8)], # row 7
        [random.randint(a=0,b=1) for x in range(0,8)]] # row 8
data

[[1, 1, 0, 0, 1, 1, 1, 0],
 [1, 1, 1, 0, 1, 0, 1, 1],
 [0, 1, 0, 1, 0, 0, 0, 0],
 [1, 1, 0, 0, 1, 1, 0, 1],
 [0, 1, 0, 1, 0, 0, 0, 0],
 [0, 1, 1, 1, 1, 1, 0, 0],
 [0, 0, 1, 0, 0, 0, 1, 1],
 [0, 0, 0, 1, 1, 0, 0, 1]]

pyplot.figure(figsize=(5,5))
pyplot.imshow(data)
pyplot.show()

pyplot.imshow(dados, alfa=0,75)
pyplot.show()

from matplotlib import colors
colormap = colors.ListedColormap(["red","green"])
pyplot.imshow(data, cmap=colormap)
pyplot.show()

pyplot.figure(figsize=(10,10))

pyplot.xlabel("x axis with ticks",
             size = 14)
pyplot.ylabel("y axis with ticks",
             size= 14)
pyplot.title("this is the title of the plot",
             size=28)

pyplot.xticks(size=14,
             color = "red")
pyplot.yticks(size=14, 
              color = "red")

colormap = colors.ListedColormap(["darkblue","lightblue"])

pyplot.imshow(data,
             cmap=colormap)

<matplotlib.image.AxesImage em 0x16408e89488>

Usarei essa abordagem para visualizar iterações em alguns estudos de simulação baseados em agentes a serem publicados em meu blog.

Linnart Felkl

Cientista de dados com foco em simulação, otimização e modelagem em R, SQL, VBA e Python