Skip to main content
 首页 » 编程设计

python之使用 Python 查找最相似的行

2025年05月04日11lyj

我有两个数据框(df1 和 df2)。在 df1 中,我用一组值存储一行,我想在 df2 中找到最相似的行。

import pandas as pd 
import numpy as np 
 
# Df1 has only one row and four columns. 
df1 = pd.DataFrame(np.array([[30, 60, 70, 40]]), columns=['A', 'B', 'C','D']) 
 
# Df2 has 50 rows and four columns 
df2 = pd.DataFrame(np.random.randint(0,100,size=(50, 4)), columns=list('ABCD')) 

问题:基于df1,df2中最相似的行是什么?

请您参考如下方法:

用df1减去df2,计算每一行的范数。找到最小范数并解决问题。

diff_df = df2 - df1.values 
# or diff_df = df2 - df1.iloc[0, :] 
norm_df = diff.apply(np.linalg.norm, axis=1) 
df2.loc[norm_df.idxmin()] 

可读性强,速度快。