Использование 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 типов сигмовидных наклонов. Есть также 10 порогов стоимости.

Я пытаюсь сделать трехмерный сюжет этого на поверхности. Чья демка это:

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 должны быть двумерными массивами.

Как я могу создать X, Y и Z, чтобы получить это в нужном формате?

Z вертикальная ось должна быть f1, а также cost_threshold а также sigmoid_slope было бы X а также Y,

Кроме того, как бы я добавил отдельный участок поверхности, где модель, скажем, no_model, а затем наложить этот участок поверхности на этот, где значения f1 колонки разные?

ОБНОВИТЬ

Я знаю, как получить 2D-массив для Z через сводную таблицу:

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

Если вы передадите их в сюжет, вы получите: