I have the following df:
print(df) document embeddings 1 [-1.1132643 , 0.793635 , 0.8664889] 2 [-1.1132643 , 0.7