Как изменить ось на тепловой карте с помощью python?

Я пытаюсь построить тепловую карту с разными значениями A. На оси показано количество итераций от 0 до 30 вместо значения A, которое должно быть от 0 до 5.

import numpy as np
from math import e
import matplotlib.pyplot as plt
from ipywidgets import interactive   
%matplotlib inline


def ricker(A, x):
    return x * e**(A * (1 - x))


import seaborn as sns

A1 = np.linspace(0,5,30)
A2 = np.linspace(0,5,30)

def F(A1,A2):

     A1,A2 = np.meshgrid(A1,A2)
     Lyap = 0
     x = 0.5
     for i in range(0,5000):
          if i % 2==0:
              A = A2
          else:
              A = A1

          Lyap = Lyap + np.log(abs((1-A*x)*e**(A - A*x)))
          x = ricker(A, x)
      Lyap = Lyap/5000
      return Lyap
 z = F(A1,A2)
 plt.figure(figsize=(16, 16))
 ax = sns.heatmap(z)
 ax.invert_yaxis()

Я получаю сообщение об ошибке с вашим кодом: NameError: name 'e' is not defined

— 07.04.2019 20:38

Извините, я включил импорт, который должен исправить эту ошибку.

— 07.04.2019 21:52

python numpy

07.04.2019 20:11

Почему в Python есть оператор "pass"?

Оператор pass в Python - это простая концепция, которую могут быстро освоить даже новички без опыта программирования.

Некоторые методы, о которых вы не знали, что они существуют в Python

Python - самый известный и самый простой в изучении язык в наши дни. Имея широкий спектр применения в области машинного обучения, Data Science,...

Основы Python Часть I

Вы когда-нибудь задумывались, почему в программах на Python вы видите приведенный ниже код?

LeetCode - 1579. Удаление максимального числа ребер для сохранения полной проходимости графа

Алиса и Боб имеют неориентированный граф из n узлов и трех типов ребер:

Оптимизация кода с помощью тернарного оператора Python

И последнее, что мы хотели бы показать вам, прежде чем двигаться дальше, это

Советы по эффективной веб-разработке с помощью Python

Как веб-разработчик, Python может стать мощным инструментом для создания эффективных и масштабируемых веб-приложений.

1 939

Перейти к ответу Данный вопрос помечен как решенный

Ответы 1

Ответ принят как подходящий

Вы должны указать метки x/y, например.

ax = sns.heatmap(z, xticklabels=A1, yticklabels=A2)

Результат довольно некрасивый

rawticklabels

Вы можете добиться большего успеха, если немного отформатируете

xlabels = ['{:3.1f}'.format(x) for x in A1]
ylabels = ['{:3.1f}'.format(y) for y in A2]
ax = sns.heatmap(z, xticklabels=xlabels, yticklabels=ylabels)

betterstrings

но вы также можете использовать меньше галочек

ax = sns.heatmap(z, xticklabels=A1, yticklabels=A2)
ax.set_xticks(ax.get_xticks()[::3])
ax.set_xticklabels(xlabels[::3])
ax.set_yticks(ax.get_yticks()[::3])
ax.set_yticklabels(ylabels[::3])

somelabels