如何使用两个列表推导创建2d numpy ndarray

Question

我尝试使用以下代码创建2D numpy ndarray：

temp = np.array([[np.mean(w2v[word]) for word in docs if word in w2v] for docs in X[:5]])

temp的形状为（5，）而不是预期的（5，x）。

临时的数据结构也类似于：array（[list（[.....]，...）]）

似乎内部列表未转换为ndarray。

Answer 1

您在那里缺少np.array，应该是：

temp = np.array([np.array([np.mean(w2v[word]) for word in docs if word in w2v] for docs in X[:5])])

运行示例：

bob
Out[70]: [[1, 2, 3], [4, 5, 6], [7, 8, 9]]

tmp = np.array([np.array([x for x in Y]) for Y in bob])

tmp
Out[72]: 
array([[1, 2, 3],
       [4, 5, 6],
       [7, 8, 9]])