我想把所有这些元素写进一个大小为381K x 381K的numpy数组的上三角。

我用过np.triuê指数. 它毁了我的记忆。

def generate_triangular_mat_gen(n_words):
    xgen_iters = chain(*[repeat(x, n_words - 1 - x) for x in range(n_words - 1)])
    ygen_iters = chain(*[range(x+1, n_words) for x in range(n_words-1)])
    return xgen_iters, ygen_iters

这种方法效果很好。但是,一次读取1个元素并复制到numpy数组位置的速度很慢。可能需要一天左右。

有没有一种方法可以一次性从迭代器中快速获取600000000个元素。这真的可以减少计算时间。