Python 3D函数图像

# -*- coding: utf-8 -*-

"""

Created on Mon Sep 10 09:51:39 2018

@author: Tanry

"""

# -*- coding: utf-8 -*-

import numpy as np

import matplotlib.pyplot as plt

from mpl_toolkits.mplot3d import Axes3D

import math

from matplotlib import cm

from matplotlib.ticker import LinearLocator, FormatStrFormatter

fig = plt.figure(figsize=(16,12))

ax = fig.gca(projection="3d")

# 准备数据

#x = np.arange(0.005, 1,0.005)    #生成[-5,5]间隔0.25的数列，间隔越小，曲面越平滑

#y = np.arange(0, 1.57,0.005)

x = np.arange(-100, 100,0.1)    #生成[-5,5]间隔0.25的数列，间隔越小，曲面越平滑

y = np.arange(-100, 100,0.1)

x, y = np.meshgrid(x,y)  #格点矩阵,原来的x行向量向下复制len(y)次，形成len(y)*len(x)的矩阵，即为新的x矩阵；原来的y列向量向右复制len(x)次，形成len(y)*len(x)的矩阵，即为新的y矩阵；新的x矩阵和新的y矩阵shape相同

#r = np.sqrt(x ** 2 + y ** 2)

#z = np.arctan(16.667*(1-1.0/x**2)*np.sin(2*y))

z = x**2+y**2+2*x*y

surf = ax.plot_surface(x, y, z, cmap=cm.coolwarm)  # cmap指color map

# 自定义z轴

#ax.set_zlim(-1.57, 1.57)

ax.set_zlim(-4000, 40000)

ax.zaxis.set_major_locator(LinearLocator(20))  # z轴网格线的疏密，刻度的疏密，20表示刻度的个数

ax.zaxis.set_major_formatter(FormatStrFormatter('%.02f'))  # 将z的value字符串转为float，保留2位小数

#设置坐标轴的label和标题

ax.set_xlabel('CS',size=15)

ax.set_ylabel('Q',size=15)

ax.set_zlabel('z',size=15)

ax.set_title("Surface plot", weight='bold', size=20)

#添加右侧的色卡条

fig.colorbar(surf, shrink=0.6, aspect=8)  # shrink表示整体收缩比例，aspect仅对bar的宽度有影响，aspect值越大，bar越窄

plt.show()