最近在优化一个代码,想要提高运算速度,有一个结构不知道怎么优化,想问下有没有人有办法。
测试用的代码如下:
import numpy as np
a=np.arange(15).reshape(5,3)
b=np.arange(15).reshape(5,3)*5
c=np.array([[0,1,3,0,2],[1,1,1,0,3],[3,2,6,7,0]])
l1=np.arange(10,15)
for i in range(2):
for k in range(0,l1[i]):
for w in range(5):
if 4-w>0.1:
Fa1=c[i,w+1]*3+10
c[i,w]=c[i,w]+Fa1
else:
c[i,w]=c[i,w]+3
c[i+1,w]=np.where(l1[i]-k<0.1,c[i,w],np.where(c[i+1,w]<10.0,0.0,c[i+1,w]))