大部分小论文三维坐标图,Z轴和默认的方向不一样,这次知道怎么做了,做个记录
主要是下面的一行代码
ax.view_init(elev=20, # 仰角
azim=45 # 方位角
)
完整代码如下
#coding:utf-8
import numpy as np
import random
from mpl_toolkits.mplot3d import Axes3D
import matplotlib.pyplot as plt
fig = plt.figure()
ax = Axes3D(fig)
ax.set_facecolor('white')
x, y, z = [], [], []
def fun(x,y):
return x+y
# demo
x = np.linspace(100,800,8)
y = np.linspace(100,800,8)
x, y = np.meshgrid(x, y)#生成网格点坐标矩阵
z = fun(x, y)
# plt.title("average accuracy")#总标题
ax.plot_surface(x, y, z, cmap=plt.cm.jet,facecolor='0.5') #cmap=plt.cm.jet控制多颜色
ax.view_init(elev=20, # 仰角
azim=45 # 方位角
)
ax.set_xlabel('enhance')
ax.set_ylabel('map')
ax.set_zlabel('accracy')#给三个坐标轴注明
# 保存成PDF放大后不失真(默认保存在了当前文件夹下)直接用Adobe编辑PDF,把灰色背景删掉
# plt.savefig('temp.pdf', bbox_inches='tight')
plt.show()