从熊猫多索引数据框创建等高线图

Question

如何从以下名为think_or_feel的熊猫数据框中创建轮廓图：

              think     feel
cNEU    cOPN        
y         n     27      20
n         n     40      23
y         y     43      25
n         y     97      63

我尝试了以下操作：

X=think_or_feel.columns
Y=think_or_feel.index 
Z=think_or_feel.values
x,y=np.meshgrid(X, Y)
plt.contourf(x,y,Z)

我收到以下错误： unhashable type: 'numpy.ndarray'

真的很感谢您的帮助。

Answer 1

我想原因是您的索引/列不是数字，而plt.contourf需要数字。让我们尝试一下：

X=np.arange(think_or_feel.shape[1])
Y=np.arange(think_or_feel.shape[0])
Z=think_or_feel.values

plt.contourf(x,y,Z)
plt.xticks(X, think_or_feel.columns)
plt.yticks(Y, think_or_feel.index)

输出：