首頁  >  文章  >  後端開發  >  如何加速Python程式

如何加速Python程式

(*-*)浩
(*-*)浩原創
2019-07-05 14:29:322468瀏覽

這次就說一種簡單的方式來加速python運算速度的方法,就是使用numba函式庫來進行,numba函式庫可以使用JIT技術即時編譯,達到高效能,另外也可以使用cuda GPU的運算能力來加速,對python來說是一個提速非常好的工具庫,使用簡單,但是安裝稍微複雜一些。

如何加速Python程式

安裝完成numba就可以使用了。 (推薦學習:Python影片教學

下面寫一個小案例來看一下加速後的程式和加速前的程式的區別,借用官網上最經典的例子:

#!/usr/bin/env python
# coding=utf-8
from numba import jit
from numpy import arange
import time

@jit
def sum2d(arr):
    M, N = arr.shape
    result = 0.0
    for i in range(M):
        for j in range(N):
            result += arr[i,j]
    return result

a = arange(9).reshape(3,3)
start_time = time.time()
for i in range(10000000):
    sum2d(a)
end_time = time.time()
print (end_time - start_time)

這裡使用numpy產生三行三列的矩陣,[[0,1,2],[3,4,5],[6,7,8]]然後做二維累加計算,值顯然應該是36,這裡做了10000000次這樣的計算,使用@jit註解可以直接的使用numba jit技術實時編譯,從而提高速度,最終運行時間大約是3.86s,如果去掉註解的話那麼運行時間大約是25.45s從這裡可以看出來大約有6.6倍的性能提升,所以使用numba加速python程式確實是方便簡單

更多Python相關技術文章,請訪問Python教程欄目進行學習!

以上是如何加速Python程式的詳細內容。更多資訊請關注PHP中文網其他相關文章!

陳述:
本文內容由網友自願投稿,版權歸原作者所有。本站不承擔相應的法律責任。如發現涉嫌抄襲或侵權的內容,請聯絡admin@php.cn