Power BI中的Python-在matplotlib.pyplot.matshow

时间:2019-04-11 07:59:58

标签: python powerbi

在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()的示例,但是我无法弄清楚如何使用它来设置文字字符串,更不用说动态列名了。

1 个答案:

答案 0 :(得分:0)

解决了这个问题...

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()