Добавление константы к оси x

Question

Я использую python и хочу сделать карту памяти с 1900 по 2000 по оси x. Итак, у меня есть двумерная функция, скажем, f (t, x), где t находится между 1900 и 2000 годами. Тогда я определяю g (x) = [f (t, x) для t в диапазоне (1900,2000)] и поместите все в массив по z = numpy.array ([g (x) для x в диапазоне (50)]). Затем я рисую тепловую карту pylab.color (z), но затем я теряю свою зависимость от лет, потому что теперь она находится в массиве, который начинается с нуля. Затем ось x равна от 0 до 100. Как добавить 1900 к оси x, а не в метку, но чтобы диапазон стал 1900-2000. Заранее спасибо.

Answer 1

Если вы используете pcolormesh (или pcolor, но pcolormesh быстрее), вам необходимо пройти в явном x и y массивов, а также в качестве z массива.

Если вы используете imshow, вам необходимо использовать extent kwarg.

В качестве примера использования pcolormesh или pcolor:

import numpy as np 
import matplotlib.pyplot as plt 

z = np.arange(40).reshape(4, 10) 
x = np.linspace(1900, 2000, 10) 
y = np.linspace(20, 50, 4) 

fig, ax = plt.subplots() 
ax.pcolormesh(x, y, z, cmap='gist_earth') 
plt.show() 

В качестве примера использования imshow (Обратите внимание, что вы должны были бы использовать aspect='auto', а также сделать imshow и pcolormesh матч точно):

import numpy as np 
import matplotlib.pyplot as plt 

z = np.arange(40).reshape(4, 10) 
x = np.linspace(1900, 2000, 10) 
y = np.linspace(20, 50, 4) 

fig, ax = plt.subplots() 
ax.imshow(z, cmap='gist_earth', interpolation='nearest', origin='lower', 
      extent=[x.min(), x.max(), y.min(), y.max()]) 
plt.show()