python繪制圖形

 　  python能快速解決日常工作中的小任務，比如數據展示。

　　python做數據展示，主要用到matplotlib庫，使用簡單的代碼，就可以很方便的繪制折線圖、柱狀圖等。使用Java等，可能還需要配合html來進行展示，十分繁瑣。

繪制的圖形舉例如下：

     

 

 

 各種平面圖的繪制代碼：

'''
File Name：   draw
Author：      tim
Date：        2018/8/15 16:47
Description： 圖形繪制。十分有用，對於工作中實驗性的項目，可以快速展示效果。如果使用java，還需要配合前端展示。
'''

import matplotlib.pyplot as plt
import numpy as np  # 模塊取別名


# 直方圖
def draw_hist():
    mu = 100
    sigma = 20

    x = mu + sigma * np.random.randn(20000)  # 樣本數量
    plt.hist(x, bins=100, color='green', normed=True)  # bins：顯示有幾個直方，normed是否對數據進行標准化

    plt._show()


# 條形圖
def draw_bar():
    y = [20, 10, 30, 25, 15]  # Y軸數據
    index = np.arange(5)  # X軸數據，也可以是index = [0,5]

    plt.bar(left=index, height=y, color='blue', width=0.5)
    plt.show()


# 折線圖
def draw_plot():
    x = np.linspace(-10, 10, 100)  # -10到10，100個點
    y = x ** 3  # x的3次冪

    plt.plot(x, y, linestyle='--', color='orange', marker='<')
    plt.xlabel('X')
    plt.ylabel('Y')

    plt.show()


# 散點圖
def draw_scatter():
    x = np.random.randn(1000)
    y = x + np.random.randn(1000) * 0.5

    plt.scatter(x, y, s=5, marker='<')  # s表示面積，marker表示圖形
    plt.show()


# 餅狀圖
def draw_pie():
    labels = 'A', 'B', 'C', 'D'  # 4個模塊
    fracs = [15, 30, 45, 10]  # 每個模塊占比例

    plt.axes(aspect=1)  # 使x、y軸比例相同
    explode = [0, 0.5, 0, 0]  # 突出某一部分區域

    plt.pie(x=fracs, labels=labels, autopct='%.0f%%', explode=explode)  # autopct顯示百分比
    plt.show()


# 帶圖例
def draw_with_legend():
    x = np.arange(1, 11, 1)  # x軸坐標，1開始，11結束，步長為1

    plt.plot(x, x * 2)  # 第一條線，x,y坐標
    plt.plot(x, x * 3)
    plt.plot(x, x * 4)

    plt.legend(['Normal', 'Fast', 'Faster'])  # 設置圖例，與上面的線對應
    plt.grid(True, color='green', linestyle='--', linewidth=1)  # 繪制網格

    plt.show()


# start
if __name__ == '__main__':
    # draw_hist()
    # draw_bar()
    draw_plot()
    # draw_scatter()
    # draw_pie()
    # draw_with_legend()

 

3D圖的繪制代碼：

'''
File Name：   draw_3d
Author：      tim
Date：        2018/8/15 18:42
Description： 3D繪圖
'''

import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D


# 3D 繪制
def draw_3D():
    fig = plt.figure()  # 定義一個窗口
    ax = Axes3D(fig)  # 繪制3D坐標

    # 設置x、y、z的值
    x = np.arange(-4, 4, 0.25)
    y = np.arange(-4, 4, 0.25)
    x, y = np.meshgrid(x, y)  # x-y 平面的網格

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

    # 做出一個三維曲面，並將一個 colormap rainbow 填充顏色，之后將三維圖像投影到 XY 平面上做一個等高線圖
    # rstride 和 cstride 分別代表 row 和 column 的跨度。
    ax.plot_surface(x, y, z, rstride=1, cstride=1, cmap=plt.get_cmap('rainbow'))

    # 添加 XY 平面的等高線
    ax.contourf(x, y, z, zdir='z', offset=-2, cmap=plt.get_cmap('rainbow'))

    ax.set_zlim(-2, 2)
    plt.show()  # 展示


# start
if __name__ == '__main__':
    draw_3D()