新闻  |   论坛  |   博客  |   在线研讨会
Python 3D函数图像
tanry111 | 2018-09-11 10:17:15    阅读:139   发布文章

# -*- 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()


参与讨论
登录后参与讨论
推荐文章
最近访客