Проблема в том, что метка Z вообще не отображается. Есть ли способ уменьшить размер коробки или увеличить фигуру, чтобы она поместилась?
Ниже приведен воспроизводимый код
import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
gt = np.random.rand(50, 96, 3)
pred_a = gt + 0.01 * np.random.randn(50, 96, 3)
pred_b = gt + 0.02 * np.random.randn(50, 96, 3)
def plot_3d_flight(gt, pred_model_a, pred_model_b, forecast_step=0):
fig = plt.figure(figsize=(10, 8))
ax = fig.add_subplot(111, projection="3d")
ax.plot(gt[:, forecast_step, 0],
gt[:, forecast_step, 1],
gt[:, forecast_step, 2],
label="Ground Truth")
ax.plot(pred_model_a[:, forecast_step, 0],
pred_model_a[:, forecast_step, 1],
pred_model_a[:, forecast_step, 2],
label="Model A")
ax.plot(pred_model_b[:, forecast_step, 0],
pred_model_b[:, forecast_step, 1],
pred_model_b[:, forecast_step, 2],
label="Model B")
ax.set_xlabel("X")
ax.set_ylabel("Y")
ax.set_zlabel("Z") #
