DataFrame：如何使用索引和列作為 x 和 y，數據作為 z 來繪制 3D 圖？

Question

我的 DataFrame 行索引和列索引存儲x和y值，而數據值是z值，即f(x, y) 。

舉個例子：

import pandas as pd
df = pd.DataFrame([[150, 120, 170], [190, 160, 130]],
                  index=[2, 4], columns=[10, 30, 70])
print(df)
#     10   30   70
# 2  150  120  170
# 4  190  160  130

那么f(4, 30)是160 。

我想做一個 3D plot 的 function f 。 我真的不介意它看起來像 3D 直方圖還是表面 plot - 這兩個答案都將不勝感激。

Answer 1

您需要從您的行和列索引創建一個網格網plot_surface meshgrid

import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
df = pd.DataFrame([[150, 120, 170], [190, 160, 130]],
                  index=[2, 4], columns=[10, 30, 70])
x,y = np.meshgrid(df.columns, df.index)
fig, ax = plt.subplots(subplot_kw={"projection": "3d"})
ax.plot_surface(x, y, df)