gpt4 book ai didi

python Pandas : Finding cosine similarity of two columns

转载 作者:太空狗 更新时间:2023-10-30 01:54:24 25 4
gpt4 key购买 nike

假设我在 python pandas.DataFrame 中有两列:

          col1 col2
item_1 158 173
item_2 25 191
item_3 180 33
item_4 152 165
item_5 96 108

获取这两列的余弦相似度的最佳方法是什么?

最佳答案

这就是你要找的吗?

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"]))

关于 python Pandas : Finding cosine similarity of two columns,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/25736861/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com