Get document vectors from spacy

spacy_doc_vectors.py

# Need to run:
# python -m spacy download en 
# from console first to get the model

import spacy
import pandas as pd


nlp = spacy.load("en")

docs = [
    "This is doc 1",
    "This is doc 2",
    "This is doc 3",
    "This is doc 4",
    "This is doc 5",
]

foo = list(nlp.pipe(docs))

dict_comp = {i.text: i.vector for i in foo}
dict_df = pd.DataFrame(dict_comp)

print(dict_df)