Python 不会加速For循环

drug_drug_adj_list = []
tmp = np.dot(drug_gene_adj, gene_drug_adj)

for i in range(n_drugdrug_rel_types):
    mat = np.zeros((n_drugs, n_drugs))
    for d1 in range(n_drugs):
        for d2 in range(n_drugs):

        if tmp[d1, d2] == i + 4:
            mat[d1, d2] = mat[d2, d1] = 1.

    drug_drug_adj_list.append(sp.csr_matrix(mat))
drug_degrees_list = [np.array(drug_adj.sum(axis=0)).squeeze() for drug_adj in drug_drug_adj_list]

n_drugdrug_rel_type 有1400个，n_drugs也有1400个 整个循环太慢了，用jit又各种报错加速啊

可以不用jit，怎么加速这段代码？

环境：python2.7 CentOS7

想办法用hash吧, 3个for本身就太慢了

您好，我是有问必答小助手，您的问题已经有小伙伴解答了，您看下是否解决，可以追评进行沟通哦~

如果有您比较满意的答案 / 帮您提供解决思路的答案，可以点击【采纳】按钮，给回答的小伙伴一些鼓励哦～～

ps:问答VIP仅需29元，即可享受5次/月 有问必答服务，了解详情>>>https://vip.csdn.net/askvip?utm_source=1146287632