Создание прямоугольной тепловой карты из двух столбцов информационного кадра Pandas

Question

У меня есть такой фрейм данных Pandas -

Значение в столбце 's' - это точность модели для соответствующих значений«к» и «ш».Таким образом, это строго между 0 и 1.

Я хочу построить тепловую карту таким образом, чтобы была создана сетка 7x2 из k (7 значений) вдоль оси X и w (2 значения) вдоль оси y и соответствующиеячейка будет окрашена в зависимости от значения s.

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

Answer 1

Попробуйте:

import matplotlib.pyplot as plt
import seaborn as sns

flights = df.pivot("w", "k", "s")
f, ax = plt.subplots(figsize=(9, 6))
sns.heatmap(flights, annot=True,  linewidths=.5, ax=ax)
plt.show()