PRP - 9 months ago 46

Python Question

I have the following script to generate a heatmap of NxN cores :

`import matplotlib.pyplot as plt`

import matplotlib.patches as mpatches

import numpy as np

matrix = np.random.randint(300,high=400, size=(4, 4))

heatmap = plt.pcolor(np.array(matrix))

for y in range(len(matrix)):

for x in range(len(matrix[0])):

plt.text(x + 0.5, y + 0.5, '%.4f' % matrix[y][x],

horizontalalignment='center',

verticalalignment='center',

)

plt.colorbar(heatmap)

plt.xlabel ("Cores")

plt.ylabel ("Cores")

plt.title ("Temperature Map of the NxN Cores")

plt.savefig ("xyz"+".png")

plt.show()

plt.clf()

As each of squares depict a core, the values of

`0.5`

`1.5`

Answer Source

You can explicitly set where you want to tick marks with `plt.xticks`

and `plt.yticks`

.

For your example:

```
plt.xticks(range(min(x), max(x)+1))
plt.yticks(range(min(y), max(y)+1))
plt.show()
```

If your data in not integers, you could use:

```
import numpy as np
plt.xticks(np.arange(min(x), max(x)+1, 1.0))
plt.yticks(np.arange(min(y), max(y)+1, 1.0))
plt.show()
```