数据框轮廓图 Python

Question

我有一个包含很多列的数据框。 我尝试绘制一个轮廓，但出现“无法将值转换为轴单位”之类的错误。

x = np.arange(len(df.index))
y = np.arange(len(df.columns))
X,Y = np.meshgrid(x,y)

cs = ax.contourf(X,Y, df.values, cmap=cm.coolwarm)
cbar = plt.colorbar(cs)
plr.show()

我究竟做错了什么？

Answer 1

列和行混合在一起。 您需要使用x作为列，使用y作为索引方向，或者转置值（ ax.contourf(X,Y, df.values.T, cmap='coolwarm') ）。

这是一个带有玩具数据的示例， x和y互换：

from matplotlib import pyplot as plt
import numpy as np
import pandas as pd

df = pd.DataFrame(np.random.rand(10, 25))
x = np.arange(len(df.columns))
y = np.arange(len(df.index))
X, Y = np.meshgrid(x, y)

fig, ax = plt.subplots()
cs = ax.contourf(X, Y, df.values, cmap='coolwarm')
plt.colorbar(cs)
plt.show()