python - 使用 Python 查找最相似的行
问题描述
我有两个数据框(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()]
它可读且快速。
推荐阅读
- android - 迁移到 androidX 后,我的应用无法连接到互联网
- gremlin - 在 AWS Neptune 上的 50M 边缘响应时间较短的 Gremlin
- javascript - 我怎样才能只允许单击我的图像?
- r - 官员::slip_in_footnote 无法让光标向前移动
- java - java.lang.IncompatibleClassChangeError:类型 com.google.gwt.dom.client.Style$Float$1 不是 com.google.gwt.dom.client.Style 的嵌套成员
- c++ - 如何从文本文件中读取 if 语句条件
- objective-c - 使用 requestDownloadFile 下载后 ICCameraDevice 删除:... 不起作用
- javascript - 从 Next JS 的公共文件夹中提供静态 JSON 文件
- java - Gradle 包装器用户名和密码在命令行中不起作用
- wordpress - 我对 Wordpress 上的 JWT 身份验证有疑问