Как можно создать трехмерный график поверхности с барами неопределенности или эквивалентными?

Question

Цель состоит в том, чтобы создать трехмерный график поверхности с полосами неопределенности или какой-либо другой четкой визуализацией неопределенностей, например:

В настоящее время у меня есть следующий график:

Генерируется с использованием следующего кода:

from io import StringIO
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
import pandas as pd
import seaborn as sns

data_string = StringIO(
    """
     V1, V2, V3, V4, V5
    110,111,109,107,108
    101,101,102,102,102
    102,102,103,103,103
    103,103,104,104,104
    100,101,100,100,100
    """
)
uncertainties_string = StringIO(
    """
    V1U,V2U,V3U,V4U,V5U
      5,  5,  5,  5,  7
      5,  5,  3,  5,  5
      6,  5,  5,  5,  5
      5,  6,  5,  2,  5
      5,  5,  5,  5,  5
    """
)

data          = pd.read_csv(data_string)
uncertainties = pd.read_csv(uncertainties_string)

df = data.unstack().reset_index()
df.columns = ["X", "Y", "Z"]

df['X'] = pd.Categorical(df['X'])
df['X'] = df['X'].cat.codes

fig = plt.figure()
ax = fig.gca(projection='3d')
surf = ax.plot_trisurf(df['Y'], df['X'], df['Z'], cmap=plt.cm.jet, linewidth=0.01)
fig.colorbar(surf)
plt.show()

Answer 1

Я не уверен, есть ли встроенная функциональность для этого, но мне было весело играть: Я делаю две полосы ошибок на одну точку данных (одну для части выше и одну для части ниже) и строю их с разными zorder:

# get errors into the same format as data
df_unc = uncertainties.unstack().reset_index()
df_unc.columns = ["X", "Y", "Z"]
df_unc['X'] = pd.Categorical(df_unc['X'])
df_unc['X'] = df_unc['X'].cat.codes

# compute lower and upper values for errorbars
df['z_low'] = df['Z'] - df_unc['Z']
df['z_high'] = df['Z'] + df_unc['Z']

# plot surface with middle value for zorder 
surf = ax.plot_trisurf(df['Y'], df['X'], df['Z'], cmap=plt.cm.jet, linewidth=0.01, zorder=2)

# plot lines with lower and higher zorder, respectively
for ix, row in df.iterrows():
    ax.plot((row['Y'], row['Y']), (row['X'], row['X']), (row['z_low'], row['Z']), c='k', zorder=1)
    ax.plot((row['Y'], row['Y']), (row['X'], row['X']), (row['Z'], row['z_high']), c='k', zorder=3)