python - 从生成器获取 numpy 数组

Question

是否正在填充一个 numpy 数组，如下所示：

import numpy as np
import bct as bct

mri_thresh = np.linspace(0.2, 0.07, 20)
W = np.random.rand(40,40)
W_bin = np.zeros([40, 40 , len(mri_thresh)])

i = 0
for thresh in mri_thresh:
    W_bin[:, :, i] = (bct.threshold(W, thresh)>0)
    i = i+1

似乎有一种更简洁的方法可以实现相同的目标

W_bin = [(bct.threshold_proportional(W, thresh)>0 for thresh in mri_thresh)]

但在这里我得到了一个发电机。我的问题是有没有办法让它产生一个 numpy 数组而不是使用类似代码的生成器？

标签： pythonnumpygenerator