Python - plotting grid based on values

Question

I checked similar threads, and my question is going to be a step further from this one: Plotting colored grid based on values

I have a grid size 20 x 10, where the first cell (bottom left) has an ID = 0 and the last one (upper right) has an ID = 99. Lets say that I have two lists. The first one is a list of cells that have a value bigger than 0 and the second list consists of those values, e.g. cell with ID = 11, has a value 77.

Cellid = [2, 4 ,5, 11 ,45 ,48 ,98]
Cellval = [20, 45 ,55, 77,45 ,30 ,15]

I would like to make a 2D plot of this grid where the cell color is based on value. Empty cells are blank, for the rest: the bigger the value, the greener the cell.
Secondly, i'd like to upgrade 2D plot from previus point with 3D-plot, where the cells values are columns.

Can you give me an advice how to approach this?

score 3 · Accepted Answer · answered Apr 04 '19 at 15:54

3

There is a contradiction between "grid size 20 x 10" and the upper right ID=99. So I will assume you mean "grid size 10 x 10" here.

You can then create an array which is 0 everywhere, except at the positions given by Cellid. Here, I assume that ID runs along x first. You can mask it to have 0 not being colorized at all; them plot it as imshow.

import numpy as np
import matplotlib.pyplot as plt

nrows = 10
ncols = 10

Cellid = [2, 4 ,5, 11 ,45 ,48 ,98]
Cellval = [20, 45 ,55, 77,45 ,30 ,15]

data = np.zeros(nrows*ncols)
data[Cellid] = Cellval
data = np.ma.array(data.reshape((nrows, ncols)), mask=data==0)

fig, ax = plt.subplots()
ax.imshow(data, cmap="Greens", origin="lower", vmin=0)

# optionally add grid
ax.set_xticks(np.arange(ncols+1)-0.5, minor=True)
ax.set_yticks(np.arange(nrows+1)-0.5, minor=True)
ax.grid(which="minor")
ax.tick_params(which="minor", size=0)

plt.show()

answered Apr 04 '19 at 15:54

ImportanceOfBeingErnest

321,279
53
665
712

I prefer this solution and yes, you were correct about my "grid mistake". To continue to second question - how would be possible to show this plot in 3D, so instead of colors, there would be a column as big as cell value? – energyMax Apr 04 '19 at 17:53
In how far would that not be answered by [the 3D bars example](https://matplotlib.org/gallery/mplot3d/3d_bars.html)? – ImportanceOfBeingErnest Apr 04 '19 at 18:24
I don't know how tp correctly define `Cellid` there in a 2D grid. – energyMax Apr 04 '19 at 18:39
You don't need to redefine Cellid. In the example, `top` is what is `data` here. You still need to adapt `x` and `y` according to `rows` and `cols`, and that's it, right? – ImportanceOfBeingErnest Apr 04 '19 at 18:42

score 1 · Answer 2 · answered Apr 04 '19 at 13:56

How about this?

x = 20
y = 10

scale_factor = 3

fig, ax = plt.subplots(figsize=(x / scale_factor, y / scale_factor))

ax.axis(xmin=0, xmax=x, ymin=0, ymax=y)
ax.tick_params(left=False, labelleft=False, bottom=False, labelbottom=False)
ax.xaxis.set_major_locator(ticker.MultipleLocator(1.0))
ax.yaxis.set_major_locator(ticker.MultipleLocator(1.0))

ax.grid(color='black')

cell_ids = [2, 4, 5, 11, 45, 48, 98]
cell_values = [20, 45, 55, 77, 45, 30, 15]

cdict = {'red':   [[0.0, 0.0, 0.0],
                   [0.5, 0.0, 0.0],
                   [1.0, 0.5, 0.5]],
         'green': [[0.0, 0.0, 0.0],
                   [0.5, 1.0, 1.0],
                   [1.0, 1.0, 1.0]],
         'blue':  [[0.0, 0.0, 0.0],
                   [0.5, 0.0, 0.0],
                   [1.0, 0.5, 0.5]]}

cmap = colors.LinearSegmentedColormap('greens', cdict)

for cell_id, cell_value in zip(cell_ids, cell_values):
    cell_x = cell_id % x
    cell_y = cell_id // y
    ax.add_artist(patches.Rectangle((cell_x, cell_y), 1.0, 1.0, color=cmap(cell_value * 255 // 100)))

(You might want to ask a separate question for the 3D part - it isn't quite clear)

Python - plotting grid based on values

2 Answers2