Python pandas: encontrar la similitud del coseno de dos columnas

Si hallas alguna incompatibilidad con tu código o proyecto, recuerda probar siempre en un entorno de testing antes añadir el código al proyecto final.

Solución:

¿Es eso lo que estás buscando?

from scipy.spatial.distance import cosine
from pandas import DataFrame


df = DataFrame("col1": [158, 25, 180, 152, 96],
                "col2": [173, 191, 33, 165, 108])

print(1 - cosine(df["col1"], df["col2"]))

También puedes usar cosine_similarity u otras métricas de similitud de sklearn.metrics.pairwise.

from sklearn.metrics.pairwise import cosine_similarity

cosine_similarity(df.col1, df.col2)
Out[4]: array([[0.7498213]])

---

---

Utiliza Nuestro Buscador