數據框輪廓圖 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()