天天看點

Matplotllib——繪制複雜函數圖與三維圖

一、繪制二維函數圖

1.1 繪制f(x)=sin2(x−2)e−x2f(x)=sin2(x−2)e−x2f(x) = sin2(x-2)e{-x^2}
代碼1:
import matplotlib.pyplot as plt
import numpy as np
plt.rcParams["font.sans-serif"]=['SimHei']  # 用于正常顯示中文标簽
plt.rcParams['axes.unicode_minus']=False    # 用來正常顯示負号

x = np.linspace(-2.5,2,256,endpoint=True)   # 繪制X軸(-2.5,2)的圖像

f =(np.sin(x-2))**2*(np.e)**(-x**2)         # y值

plt.plot(x,f,"g-",lw=2.5,label="f(x)")
plt.title('f(x) = sin^2(x-2)e^{-x^2}函數圖')
plt.legend()
plt.show()

_______________________________________________
代碼2(對于複雜函數,我們可以将其拆分成多單一函數)
        分析:sin^2(x-2)e^{-x^2}分解:
             sin(x1)*sin(x1)*e^x2
        其中:x1 = x-2;    x2 = -x^2
import matplotlib.pyplot as plt
import numpy as np
plt.rcParams['font.sans-serif'] = ['SimHei']  # 用來正常顯示中文标簽 
plt.rcParams['axes.unicode_minus'] = False    # 用來正常顯示負号 

x = np.linspace(0, 2,300, endpoint=True) 
x_1 = x-2
S_1 = np.sin(x_1)
S_2 = S_1**2

E_1 = -x**2
E_2 = np.exp(E_1)
f = S_2*E_2    
plt.figure(figsize = ((8,6)))     # 設定畫布大小(可省略)

plt.plot(x,f,'b-',lw=2.5,label='f(x)=sin^2(x-2)e^{-x^2}') 

plt.legend()                      # 顯示圖例       
Matplotllib——繪制複雜函數圖與三維圖
1.2 、繪制 sigmoid函數圖:f(x)=sin2(x−2)e−x2f(x)=sin2(x−2)e−x2f(x) = sin2(x-2)e{-x^2}
import matplotlib.pyplot as plt
import numpy as np
plt.rcParams['font.sans-serif'] = ['SimHei']  # 用來正常顯示中文标簽 
plt.rcParams['axes.unicode_minus'] = False    # 用來正常顯示負号 

x = np.linspace(-10, 10,300, endpoint=True) 

E_1 = -x
E_2 = np.exp(E_1)
f = 1/(1+E_2)    
plt.figure(figsize = ((8,6)))     # 設定畫布大小(可省略)
ax1 = plt.subplot(111)

plt.plot(x,f,'b-',lw=2.5,label='f(x)=\\frac{1}{1+e^{-x}}') 

plt.legend()                      # 顯示圖例 
ax1.grid(True)                    # 顯示網格       
Matplotllib——繪制複雜函數圖與三維圖
1.3、繪制正态分布圖

其中 s 為 σ

σ

, m為 μ

μ

import math
import matplotlib.pyplot as plt
import numpy as np
def gd(x,m,s):                       #其中s為sigma ,m為 mu
    left=1/(math.sqrt(2*math.pi)*s)
    right=math.exp(-math.pow(x-m,2)/(2*math.pow(s,2)))
    return left*right
def showfigure():
    plt.figure(figsize = ((8,6)))   #設定畫布大小(可省略)
    x=np.arange(-4,5,0.1)           #繪制x(-4,5)
    y=[]
    for i in x:
        y.append(gd(i,0,1))         #m為0,s為1
    plt.plot(x,y) 
    plt.xlim(-4.0,5.0)
    plt.ylim(-0.2,0.5)
    ax = plt.gca()
    ax.spines['right'].set_color('none')
    ax.spines['top'].set_color('none')
    ax.xaxis.set_ticks_position('bottom')
    ax.spines['bottom'].set_position(('data',0))
    ax.yaxis.set_ticks_position('left')
    ax.spines['left'].set_position(('data',0))

    #設定并添加标簽
    label_f1 = "$\mu=0,\ \sigma=1$"
    plt.text(2.5,0.3,label_f1,fontsize=15,verticalalignment="top",
            horizontalalignment="left")
    label_f2 = r"$f(x)=\frac{1}{\sqrt{2\pi}\sigma}exp(-\frac{(x-\mu)^2}{2\sigma^2})$"
    plt.text(1.5,0.4,label_f2,fontsize=15,verticalalignment="top"
            ,horizontalalignment="left")
    plt.show()

def main():      
Matplotllib——繪制複雜函數圖與三維圖

二、繪制三維圖

2.1 繪制三維螺旋圖
from mpl_toolkits.mplot3d import Axes3D
import numpy as np
import matplotlib.pyplot as plt

fig = plt.figure(figsize = ((8,6)))
ax = fig.add_subplot(1,1,1,projection='3d')
theta = np.linspace(-4 * np.pi, 4 * np.pi, 500) # theta旋轉角從-4pi到4pi,相當于兩圈
z = np.linspace(0, 2, 500)  # z軸從下到上,從-2到2之間畫100個點
r = z                       # 半徑設定為z大小
x = r * np.sin(theta)       # x和y畫圓
y = r * np.cos(theta)       # x和y畫圓
ax.plot(x, y, z, label='curve')
ax.legend()                 # 圖例      
Matplotllib——繪制複雜函數圖與三維圖
2.2 繪制三維線性點圖
from mpl_toolkits.mplot3d import Axes3D
import numpy as np
import matplotlib.pyplot as plt

fig = plt.figure()
ax = fig.add_subplot(1,1,1,projection='3d')
x = np.linspace(0, 5, 10)      # 在0-5之間生成10個點的向量
y = np.linspace(0, 5, 10)      # 在0-5之間生成10個點的向量
z1 = x
z2 = 2*x
z3 = 3*x
ax.scatter(xx, yy, zz1, c='red', marker='o')   # o型符号
ax.scatter(xx, yy, zz2, c='green', marker='^') # 三角型符号
ax.scatter(xx, yy, zz3, c='black', marker='*') # 星型符号
ax.legend()                                    # 顯示圖例        
Matplotllib——繪制複雜函數圖與三維圖
2.3 繪制三維柱狀圖
import random
import matplotlib as mpl
import matplotlib.dates as mdates
from mpl_toolkits.mplot3d import Axes3D
plt.rcParams['font.sans-serif'] = ['SimHei']  # 用來正常顯示中文标簽 
plt.rcParams['axes.unicode_minus'] = False    # 用來正常顯示負号 

mpl.rcParams['font.size'] = 10

fig = plt.figure(figsize=((8,6)))
ax = fig.add_subplot(111, projection='3d')
for z in [2011, 2012, 2013, 2014]:
    xs = range(1,13)
    ys = 1000 * np.random.rand(12)
    color = plt.cm.Set2(random.choice(range(plt.cm.Set2.N)))
    ax.bar(xs, ys, zs=z, zdir='y', color=color, alpha=0.8)

ax.xaxis.set_major_locator(mpl.ticker.FixedLocator(xs))
ax.yaxis.set_major_locator(mpl.ticker.FixedLocator(ys))
ax.set_xlabel('月份')
ax.set_ylabel('年份')
ax.set_zlabel('銷售額 ')
plt.show()      
Matplotllib——繪制複雜函數圖與三維圖
2.4 繪制三維 鞍部 曲面圖
import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
from matplotlib import cm

n_angles = 1000                #曲面銜接角度(平滑度)
n_radii = 20                   #鞍部半徑(1:銳角,20:平滑角)
fig = plt.figure(figsize=((10,10)))
radii = np.linspace(0.125, 1.0, n_radii)
angles = np.linspace(0, 2 * np.pi, n_angles, endpoint=False)
angles = np.repeat(angles[..., np.newaxis], n_radii, axis=1)

x = np.append(0, (radii * np.cos(angles)).flatten())
y = np.append(0, (radii * np.sin(angles)).flatten())

z = np.sin(-x * y)
fig = plt.figure()
ax = fig.gca(projection='3d')
ax.plot_trisurf(x, y, z,   
                cmap=cm.jet,    #曲面顔色
                linewidth=0.2)      
Matplotllib——繪制複雜函數圖與三維圖
2.5 繪制山區圖 f(x)=sin2(x−2)e−x2f(x)=sin2(x−2)e−x2f(x) = sin2(x-2)e{-x^2}
import numpy as np
import matplotlib.pyplot as plt
import mpl_toolkits.mplot3d

x,y = np.mgrid[-2:2:100j,-2:2:100j] #100j為設定曲面平滑度
z=x*np.exp(-x**2-y**2)
ax = plt.subplot(111, projection='3d')
ax.set_title('山區圖');
ax.plot_surface(x,y,z,rstride=2, cstride=1, cmap=cm.jet)

ax.set_xlabel('X')                #設定坐标軸标簽
ax.set_ylabel('Y')
ax.set_zlabel('Z')
plt.show()      
Matplotllib——繪制複雜函數圖與三維圖
2.6 三維動态圖
from pyecharts import Bar3D
bar3d = Bar3D("3D 柱狀圖示例", width=1200, height=600)

x_axis = ["12a", "1a", "2a", "3a", "4a", "5a", "6a", "7a", "8a", "9a", "10a", "11a",
          "12p", "1p", "2p", "3p", "4p", "5p", "6p", "7p", "8p", "9p", "10p", "11p"]
y_axis = [ "Saturday", "Friday", "Thursday", "Wednesday", "Tuesday", "Monday", "Sunday"]

data = [
    [0, 0, 5], [0, 1, 1], [0, 2, 0], [0, 3, 0], [0, 4, 0], [0, 5, 0],
    [0, 6, 0], [0, 7, 0], [0, 8, 0], [0, 9, 0], [0, 10, 0], [0, 11, 2],
    [0, 12, 4], [0, 13, 1], [0, 14, 1], [0, 15, 3], [0, 16, 4], [0, 17, 6],
    [0, 18, 4], [0, 19, 4], [0, 20, 3], [0, 21, 3], [0, 22, 2], [0, 23, 5],
    [1, 0, 7], [1, 1, 0], [1, 2, 0], [1, 3, 0], [1, 4, 0], [1, 5, 0],
    [1, 6, 0], [1, 7, 0], [1, 8, 0], [1, 9, 0], [1, 10, 5], [1, 11, 2],
    [1, 12, 2], [1, 13, 6], [1, 14, 9], [1, 15, 11], [1, 16, 6], [1, 17, 7],
    [1, 18, 8], [1, 19, 12], [1, 20, 5], [1, 21, 5], [1, 22, 7], [1, 23, 2],
    [2, 0, 1], [2, 1, 1], [2, 2, 0], [2, 3, 0], [2, 4, 0], [2, 5, 0],
    [2, 6, 0], [2, 7, 0], [2, 8, 0], [2, 9, 0], [2, 10, 3], [2, 11, 2],
    [2, 12, 1], [2, 13, 9], [2, 14, 8], [2, 15, 10], [2, 16, 6], [2, 17, 5],
    [2, 18, 5], [2, 19, 5], [2, 20, 7], [2, 21, 4], [2, 22, 2], [2, 23, 4],
    [3, 0, 7], [3, 1, 3], [3, 2, 0], [3, 3, 0], [3, 4, 0], [3, 5, 0],
    [3, 6, 0], [3, 7, 0], [3, 8, 1], [3, 9, 0], [3, 10, 5], [3, 11, 4],
    [3, 12, 7], [3, 13, 14], [3, 14, 13], [3, 15, 12], [3, 16, 9], [3, 17, 5],
    [3, 18, 5], [3, 19, 10], [3, 20, 6], [3, 21, 4], [3, 22, 4], [3, 23, 1],
    [4, 0, 1], [4, 1, 3], [4, 2, 0], [4, 3, 0], [4, 4, 0], [4, 5, 1],
    [4, 6, 0], [4, 7, 0], [4, 8, 0], [4, 9, 2], [4, 10, 4], [4, 11, 4],
    [4, 12, 2], [4, 13, 4], [4, 14, 4], [4, 15, 14], [4, 16, 12], [4, 17, 1],
    [4, 18, 8], [4, 19, 5], [4, 20, 3], [4, 21, 7], [4, 22, 3], [4, 23, 0],
    [5, 0, 2], [5, 1, 1], [5, 2, 0], [5, 3, 3], [5, 4, 0], [5, 5, 0],
    [5, 6, 0], [5, 7, 0], [5, 8, 2], [5, 9, 0], [5, 10, 4], [5, 11, 1],
    [5, 12, 5], [5, 13, 10], [5, 14, 5], [5, 15, 7], [5, 16, 11], [5, 17, 6],
    [5, 18, 0], [5, 19, 5], [5, 20, 3], [5, 21, 4], [5, 22, 2], [5, 23, 0],
    [6, 0, 1], [6, 1, 0], [6, 2, 0], [6, 3, 0], [6, 4, 0], [6, 5, 0],
    [6, 6, 0], [6, 7, 0], [6, 8, 0], [6, 9, 0], [6, 10, 1], [6, 11, 0],
    [6, 12, 2], [6, 13, 1], [6, 14, 3], [6, 15, 4], [6, 16, 0], [6, 17, 0],
    [6, 18, 0], [6, 19, 0], [6, 20, 1], [6, 21, 2], [6, 22, 2], [6, 23, 6]
    ]
range_color = ['#313695', '#4575b4', '#74add1', '#abd9e9', '#e0f3f8', '#ffffbf',
               '#fee090', '#fdae61', '#f46d43', '#d73027', '#a50026']
bar3d.add("", x_axis, y_axis, [[d[1], d[0], d[2]] for d in data],
          is_visualmap=True, visual_range=[0, 20],
          visual_range_color=range_color, 
          grid3d_width=200, grid3d_depth=80,
          grid3d_shading='lambert',             #柱狀圖顯示更真實(可省略)
          is_grid3d_rotate=True,                #自動旋轉(可省略)
          grid3d_rotate_speed=10)               #旋轉速度(可省略)
bar3d.render()