from sklearn.metrics.pairwise import cosine_similarity
import pandas as pd
import numpy as np

df = pd.DataFrame({"col1": [1,2,3], "col2": [4,5,6]})
query = np.mat([1,3])
print(cosine_similarity(df, query))

【 在 kpo (kpkpkp) 的大作中提到: 】
: 道理是这个道理
: 不过有更成熟的库来解决，但这个库我还用不太明白……
--
FROM 125.78.66.*