тепловая карта морского дна из базы данных панд с NaNs - PullRequest
Новая
       17

тепловая карта морского дна из базы данных панд с NaNs

0 голосов
/ 16 декабря 2018

Привет! Я действительно хочу создать тепловую карту, но изо всех сил: 

# correlations between undergrad studies and occupation
data_uni = n.groupby(['Q5','Q6'])['Q6'].count().to_frame(name = 'count').reset_index()
# some participants did not answer the question in the survey
data_uni.fillna('Unknown', inplace=True)

data_uni.pivot(index='Q5', columns='Q6', values='count')
plt.figure(1, figsize=(14,10))
sns.heatmap(data_uni, cmap="YlGnBu")

Я получаю сообщение об ошибке: "TypeError: ufunc 'isnan' не поддерживается для типов ввода и входовне может быть безопасно приведен к каким-либо поддерживаемым типам в соответствии с правилом приведения «safe» ».

Это правильный способ создания тепловой карты?Если да, что я делаю не так, и если нет, то что будет правильным?Спасибо за помощь!

1 Ответ

0 голосов
/ 16 декабря 2018

В соответствии с выпуском GH375 , вы можете указать маску, в которой данные не будут отображаться для тех ячеек, значения маски которых True. 

sns.heatmap(data_uni, cmap="YlGnBu", mask=data_uni.isnull())
...