使用 Python 匹配数据质量,从文件1挑选接近文件2的数据

本代码使用 Python 的 Pandas 库,根据文件2中数据的'质量',从文件1中筛选出质量最接近的数据。

代码如下:

import pandas as pd

# 读取文件1和文件2
file1 = pd.read_csv('file1.csv')
file2 = pd.read_csv('file2.csv')

# 对文件1按照质量从小到大排序
file1_sorted = file1.sort_values(by=['质量'])

# 遍历文件2中的每一行,找到文件1中离该行质量最近的数据
for i in range(len(file2)):
    # 获取该行质量
    mass = file2.loc[i, '质量']
    # 在文件1_sorted中找到离该行质量最近的数据
    closest_mass = file1_sorted.iloc[(file1_sorted['质量']-mass).abs().argsort()[:1]].iloc[0]
    # 打印该行数据和最接近的数据的位置和质量
    print(f'文件2中第{i+1}行数据:位置-{file2.loc[i, '位置']},质量-{mass}')
    print(f'最接近的文件1中的数据:位置-{closest_mass['位置']},质量-{closest_mass['质量']}')

代码说明:

  1. 首先使用 pd.read_csv() 函数读取文件1和文件2。
  2. 使用 sort_values() 函数将文件1按照'质量'列从小到大排序,方便后续查找。
  3. 遍历文件2中的每一行数据,获取该行的'质量'值。
  4. 在文件1_sorted中使用 (file1_sorted['质量']-mass).abs().argsort()[:1] 找出与当前质量值最接近的数据行。
  5. 打印出文件2中当前行数据的位置和质量,以及文件1中找到的最接近数据的位置和质量。

该代码可以帮助用户快速找到两个文件中质量相近的数据,并方便地进行后续分析或操作。

Python 代码:根据质量匹配数据,从文件1中挑选接近文件2质量的数据

原文地址: https://www.cveoy.top/t/topic/npUv 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录