如何解决提高使用if-else语句编写的重叠分段函数的计算速度
从许多.json文件的目录中读取m。
我为此编写的Python代码如下:
for i in os.listdir(PATH):
if i.endswith(".json"):
# Read in m value from json files
posterior_files = bilby.result.read_in_result(PATH + "/" + i)
m = np.asarray(posterior_files.posterior['mass_1'].values)
p = np.asarray(posterior_files.posterior['log_prior'].values)
m = [] # one big float array per each json file
A = []
B = []
C = []
for j in range(0,len(m)):
if m1[j]<5. and m1[j]>10.:
A = 0
else:
A = m**(-1.) # some algebraic expression
if m1[j]<3. and m1[j]>10.:
B = 0
else:
B = m**(-2.) # some algebraic expression
if m1[j]<5. and m1[j]>10.:
C = 0
else:
C = m**(-3.) # some algebraic expression
num = 1.0/len(m) * np.sum((A+B)/np.exp(p))
den = 1.0/len(m) * np.sum((C)/np.exp(p))
factor = num/den
print(factor)
(我省略了真实的代数表达式,因为它对问题本身并不重要。)
这段代码可以运行,但是问题是运行时间很长。有了实际的数据和表达式,花了一天的时间在超级计算机上运行。我需要重写它,以便花费更少的时间。
我将非常感谢您的任何建议!
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。