获取与数据框中第一行最接近的top_n行

import pandas as pd
import numpy as np

# 创建示例数据框
data = {'A': [1, 2, 3, 4, 5],
        'B': [0.5, 0.7, 0.3, 0.9, 0.2],
        'C': [10, 5, 8, 3, 6]}
df = pd.DataFrame(data)

# 计算第一行与其他行之间的欧氏距离
distances = np.sqrt(np.sum((df.iloc[0] - df.iloc[1:]) ** 2, axis=1))

# 将距离存储在新的列中
df['Distance'] = distances

# 按照距离列进行排序
df = df.sort_values('Distance')

# 获取与第一行最接近的前3行
top_n = 3
closest_rows = df.iloc[1:top_n+1]

print(closest_rows)