Как объединить элементы для HeatMap или TreeMap? - PullRequest
Новая
       18

Как объединить элементы для HeatMap или TreeMap?

0 голосов
/ 10 января 2020

У меня есть простой фрейм данных, который выглядит следующим образом. 

Rating  Count
AA+ 0.06
AA  0
AA  0
AA+ 0.09
BBB 0.55
AA+ 0.04
AA  0
AA  0.11
B   0.54
A   0
BBB 0.5

Я попытался построить его как HeatMap и TreeMap. Я протестировал этот простой код и получил ошибку. 

f, ax = plt.subplots(figsize=(9, 6))
sns.heatmap(df1, annot=True, fmt=".01", linewidths=.5, ax=ax)

Ошибка: TypeError: ufunc 'isnan' not supported for the input types, and the inputs could not be safely coerced to any supported types according to the casting rule ''safe''

Я думаю, что фрейм данных должен быть повернут или каким-то образом приведен в другую форму. То, что я хотел бы сделать, это подсчитать количество вхождений «AA + 0,06» и «AA + 0,04» и др. c. Есть несколько повторяющихся шаблонов. Там может быть 1, 2, 3, 4 и т. Д. c., Количество или каждый. Как я могу это сделать и отобразить результаты в HeatMap или TreeMap? Спасибо.

1 Ответ

0 голосов
/ 11 января 2020

Вот что у меня сработало. 

import squarify    # pip install squarify (algorithm for treemap)
import matplotlib

...etc.
squarify.plot(sizes=df_rtg['Count'], label=df_rtg['Rating'], alpha=.8 )
plt.axis('off')
plt.show()
...