社区所有版块导航
Python
python开源   Django   Python   DjangoApp   pycharm  
DATA
docker   Elasticsearch  
aigc
aigc   chatgpt  
WEB开发
linux   MongoDB   Redis   DATABASE   NGINX   其他Web框架   web工具   zookeeper   tornado   NoSql   Bootstrap   js   peewee   Git   bottle   IE   MQ   Jquery  
机器学习
机器学习算法  
Python88.com
反馈   公告   社区推广  
产品
短视频  
印度
印度  
Py学习  »  Python

如何加速python中的脚本代码?

John • 5 年前 • 1629 次点击  

我花了很多时间来加速下面的脚本,但我不能。你能帮帮我还是建议加快速度?这是我的密码

import numpy as np
import time
#Unchange
input = np.random.randn(300,400,500)
step_x = 30
step_y = 40
step_z = 50
lst_x = np.arange(0, input.shape[0] + step_x, step_x)
lst_y = np.arange(0, input.shape[1] + step_y, step_y)
lst_z = np.arange(0, input.shape[2] + step_z, step_z)
results = np.zeros((input.shape))
t0 = time.time()
for i in range(len(lst_x)):
  for j in range(len(lst_y)):
      for k in range(len(lst_z)):
          xx= lst_x[i]
          yy= lst_y[j]
          zz= lst_z[k]
          patch = input[xx : xx + step_x, yy : yy + step_y, zz : zz + step_z]          
          patch_added = np.random.randn(patch.shape[0],patch.shape[1],patch.shape[2]) + patch
          results[xx : xx + step_x, yy : yy + step_y, zz : zz + step_z] += patch_added
print ('Time consumption: ', time.time()-t0)

你可以在网上跑步 https://repl.it/repls/PleasingGhostwhiteNetworking

目前,repl.it需要11秒。你能在python 3中使它更快吗?

更新: 我有一个原始的数组输入是3d数组。从数组中,我将提取大小为30x40x50的补丁并处理该补丁。之后,将处理的修补程序分配到结果数组中,其位置与 input

Python社区是高质量的Python/Django开发社区
本文地址:http://www.python88.com/topic/46934
 
1629 次点击  
文章 [ 1 ]  |  最新文章 5 年前
AKX
Reply   •   1 楼
AKX    6 年前

改变

patch_added = np.random.randn(patch.shape[0],patch.shape[1],patch.shape[2]) + patch

patch_added = np.zeros(patch.shape) + patch

将“时间消耗”减少到大约2秒,而不是10-11秒。

这意味着脚本大部分时间都在生成随机数。