显示一维形状的二维 numpy ndarray

Question

I am getting a numpy array from the following function

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)

after the calling this function

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

Now, I was expecting shape of X is (18,2) but it showing (18, 1) as shown in following image.

I think I have done something wrong somewhere. Would anyone please correct me. I would be thankful. thank you.

Answer 1

If you upload the matrix X or the DataFrame , I could provide a more details answer.

In the function you use word_vec.append(token.vector_norm) so you append elements into a list.

and then you do: np.asarray(word_vec) , ie you convert the list into numpy array

This lead to the observed output.