想在一下代码生成的三维图中插入图例,请问该怎么写代码
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
import matplotlib as mpl
mpl.rcParams['font.sans-serif'] = ['SimHei'] # 指定默认字体
mpl.rcParams['axes.unicode_minus'] = False # 解决保存图像是负号'-'显示为方块的问题
from mpl_toolkits.mplot3d import Axes3D #绘制三维曲线图
import matplotlib; matplotlib.use('TkAgg')
from scipy.stats import weibull_min
T=24
file= pd.read_csv('经典场景.csv',encoding='gbk')
PW1=file['场景一']
PW2=file['场景二']
PW3=file['场景三']
PW4=file['场景四']
PV1=file['场景1']
PV2=file['场景2']
PV3=file['场景3']
PV4=file['场景4']
#图3 风电典型场景集
fig3=plt.figure(figsize=(8,8))#指定画布大小
ax3=fig3.gca(projection='3d')#指定为3D
for i in range(4):#遍历每一个场景
ax3.plot(11np.ones(T),np.arange(T), PW1)
ax3.plot(12np.ones(T),np.arange(T), PW2)
ax3.plot(13np.ones(T),np.arange(T), PW3)
ax3.plot(14np.ones(T),np.arange(T), PW4)
ax3.set_xlabel('Scenes')
ax3.set_ylabel('Time')
ax3.set_zlabel('Wind power output/kW')
plt.legend()
plt.yticks(np.arange(0,T,1)+1)
plt.title('Wind power classic scene set')
plt.show()
Python三维图像的绘制以及相关图例添加(3D图像 - 图例)
https://blog.csdn.net/LiangWF22/article/details/126282721