Блог

У меня есть набор данных, аналогичный приведенному ниже, в Power BI:

 last_updated    product     price
01-01-2019      Cycle       1000
02-01-2019      Cycle       1010
01-01-2019      Helmet      200
02-01-2019      Helmet      190
  

По сути, я хотел динамически позволить пользователю выбирать некоторые продукты из фильтров, и я бы заставил Python сводить данные и строить корреляционную матрицу.

Это всего лишь мой второй день с Python, но мне удалось написать следующий код.

 dataset = dataset.pivot(
    index = 'last_updated',
    columns = 'product',
    values = 'price'
)

matplotlib.pyplot.matshow(dataset.corr('pearson'))

matplotlib.pyplot.show()
  

Он работает, как и ожидалось, но он показывает 0, 1, 2 и т.д. Вместо Cycle, Helmet, …

Как я могу динамически устанавливать метки для имен столбцов?

Я вижу, что в некоторых примерах используется set_xticklabels() , но я не могу понять, как использовать его для задания буквенной строки, не говоря уже о динамических именах столбцов.

Решил это так…

 import seaborn as sns

dataset = dataset.pivot(
    index = 'last_updated',
    columns = 'symbol',
    values = 'cumulative_return'
)

corr = dataset.corr('pearson')
sns.heatmap(corr, annot=True, xticklabels=corr.columns, yticklabels=corr.columns)
matplotlib.pyplot.show()