Масштабирование оси в Matplotlib imshow при уникальном вызове функции

Я написал функциональный модуль, который принимает аргумент двух переменных. Для построения у меня было

x, y = pylab.ogrid[0.3:0.9:0.1, 0.:3.5:.5]
z = np.zeros(shape=(np.shape(x)[0], np.shape(y)[1]))

for i in range(len(x)):
    for j in range(len(y[0])):    
        z[i][j] = fancyFunction(x[i][0], y[0][j])

pylab.imshow(z, interpolation="gaussian")

Я получаю следующее изображение:

Но когда я попытался масштабировать оси x и y, чтобы они соответствовали диапазонам [0,3 :0,9 :0,1, 0, :3,5 :0,5] через pylab.imshow(z, interpolation="gaussian", extent=[.3,.9,0.,3.5]), я получаю

Я часами гуглил, но так и не смог найти способ сделать квадратный график с разным масштабом оси.

Спасибо!