使用mplot3D繪制DataFrame

Question

我有一個這樣的數據框：

     f1        model  cost_threshold  sigmoid_slope
366  0.140625  open          0.0001         0.0001
445  0.356055  open          0.0001         0.0010
265  0.204674  open          0.0001         0.0100
562  0.230088  open          0.0001         0.0500
737  0.210923  open          0.0001         0.1500
117  0.161580  open          0.0001         0.1000
763  0.231648  open          0.0001         0.3000
466  0.186228  open          0.0001         0.5000
580  0.255686  open          0.0001         0.7500
520  0.163478  open          0.0001         1.0000
407  0.152488  open          0.0010         0.0001
717  0.183946  open          0.0010         0.0010
708  0.201499  open          0.0010         0.0100
570  0.179720  open          0.0010         0.0500
722  0.200326  open          0.0010         0.1500
316  0.187692  open          0.0010         0.1000
240  0.243612  open          0.0010         0.3000
592  0.274322  open          0.0010         0.5000
254  0.309560  open          0.0010         0.7500
400  0.225460  open          0.0010         1.0000
148  0.494311  open          0.0100         0.0001
100  0.498199  open          0.0100         0.0010
155  0.473008  open          0.0100         0.0100
494  0.484625  open          0.0100         0.0500
754  0.504391  open          0.0100         0.1500
636  0.425798  open          0.0100         0.1000
109  0.446701  open          0.0100         0.3000
759  0.509829  open          0.0100         0.5000
345  0.522837  open          0.0100         0.7500
702  0.511971  open          0.0100         1.0000

有更多的塊，但是如您所見，每個cost_threshold包含10種S型斜率。 還有10個成本閾值。

我正在嘗試根據此處的表面圖制作3D圖。 誰的演示是：

from mpl_toolkits.mplot3d import axes3d
import matplotlib.pyplot as plt
import numpy as np

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
X, Y, Z = axes3d.get_test_data(0.05)
ax.plot_wireframe(X, Y, Z, rstride=10, cstride=10)

plt.show()

X，Y和Z必須是2D數組。

如何創建需要的X，Y和ZI格式？

垂直軸Z應該為f1 ， cost_threshold和sigmoid_slope為X和Y

另外，我如何添加一個單獨的表面圖（模型no_model ，然后將此表面圖覆蓋到該表面圖（ f1列的值不同）呢？

更新

我知道如何通過數據透視表獲取Z的2D數組：

Z = df.pivot_table('f1', 'cost_threshold', 'sigmoid_slope', fill_value=0).as_matrix()

仍然不知道如何為X和Z創建一個。

Answer 1

這是分別獲得X，Y和Z的方法：

Z = df.pivot_table('f1', 'cost_threshold', 'sigmoid_slope', fill_value=0).as_matrix()

Y = df.groupby("cost_threshold").sigmoid_slope.apply(pd.Series.reset_index, drop=True).unstack().values

Z = df.groupby("sigmoid_slope").cost_threshold.apply(pd.Series.reset_index, drop=True).unstack().values

如果將它們傳遞到情節中，則會得到：