显示一维形状的二维二维numpy ndarray

Question

我从以下函数中获取了一个numpy数组

def get_word_vec_norm(post):
    doc = nlp(post)
    word_vec = []
    index = 1
    for token in doc:
        word_vec.append(token.vector_norm)
        index = index + 1
        if index>2:
            break
#     return word_vec
    return np.asarray(word_vec)

调用此函数后

X = data['cleaned_post'].apply(get_word_vec_norm)

现在，我期望X的形状为(18,2)，但它显示(18, 1)，如下图所示。

我认为我在某处做错了。任何人都可以纠正我。我会很感激。谢谢。

Answer 1

如果您上载矩阵 for (int r = 0; r < totalRow; r++) { 或X，我可以提供更多详细的答案。

在函数中使用DataFrame，以便将元素追加到列表中。

然后执行：word_vec.append(token.vector_norm)，即将列表转换为numpy数组

这导致观察到的输出。