Тепловая карта: взять ось из индексов фреймов данных

Question

Я переименовал индексы фрейма данных A и построил его как тепловую карту:

import pandas as pd
import numpy as np
import matplotlib.pyplot as plt

A = pd.DataFrame(np.array([[1,2,3,4],[5,6,7,8]]))

columns = np.array([ 0.   ,  0.635,  1.27 ,  1.905])
rows = np.array([ 0.   ,  0.635])

A.columns = columns
A.index = rows

plt.imshow(A)

введите описание изображения здесь

Ось x графика показывает значения фрейма данных в точках 0, 1, 2, 3 и оси Y в точках 0, 1

Как я могу изменить значения оси к осям фрейма данных?

В хотите, чтобы значения по оси x были на 0, 0,635, 1,27, 1,905 и оси y на 0, 0,635

Answer 1

На мой взгляд, Seaborn предпочтительнее matplotlib.pyplot для создания тепловых карт.

Если вы хотите использовать seabobrn

import seaborn as sns 
sns.heatmap(A, yticklabels = rows, xticklabels = columns)  
plt.show() 

введите описание изображения здесь