显示1-d形状的2-d 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

如果您上载矩阵X或DataFrame，我可以提供更多详细的答案。

在函数中，您使用word_vec.append(token.vector_norm)，因此可以将元素追加到列表中。