如何在两列中都没有出现的项目的 pandas 中绘制热图 [英] How to draw a heatmap in pandas with items that don't occur in both columns

问题描述

在

In How to draw a graphical count table in pandas I asked how to draw a heatmap from input data such as:

customer1,customer2
a,b
a,c
a,c
b,a
b,c
b,c
c,c
a,a
b,c
b,c

The answer was

x = df.pivot_table(index='customer1',columns='customer2',aggfunc='size',fill_value=0)
idx = x.max(axis=1).sort_values(ascending=0).index
sns.heatmap(x[idx].reindex(idx), annot=True)

This gives a square matrix showing the number of counts for each pair from the two columns.

This solution doesn't work however if there items in the first column which don't appear in the second. For example:

a,b
a,c
c,b

Gives an error saying that [u,'a'] is not in the Index.

Is there a simple solution?

解决方案

Try this:

In [129]: df
Out[129]:
  customer1 customer2
0         a         b
1         a         c
2         a         c
3         b         b
4         b         c
5         b         c
6         c         c
7         a         b
8         b         c
9         b         c

In [130]: x = df.pivot_table(index='customer1',columns='customer2',aggfunc='size',fill_value=0)

In [131]: idx = x.max(axis=1).sort_values(ascending=0).index

In [132]: cols = x.max().sort_values(ascending=0).index

In [133]: sns.heatmap(x[cols].reindex(idx), annot=True)
Out[133]: <matplotlib.axes._subplots.AxesSubplot at 0xbb22588>

这篇关于如何在两列中都没有出现的项目的 pandas 中绘制热图的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！