You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

基于XY坐标属性计算对象间距离:患者与医院匹配需求

问题解决:为患者匹配最近医院及医院距离排序

问题背景

现有患者(Patient)和医院(Hospital)两类对象的初始化代码如下,已创建包含25名患者的PatientList与5家医院的HospitalList,需要实现:

  • 为每位患者找到最近的医院
  • 支持按距离对医院排序,可获取次近、第三近等顺位的医院

原初始化代码:

import random

class Patient:
    def __init__(self):
        x = random.randint(0,100)
        y = random.randint(0,100)

class Hospital:
    def __init__(self):
        x = random.randint(0,100)
        y = random.randint(0,100)

创建列表的代码:

PatientList = []
for p in range(25):
    PatientList.append(Patient())

HospitalList = []
for p in range(5):
    HospitalList.append(Hospital())

解决方案

1. 修复类的属性访问问题

原代码中x和y是初始化方法内的局部变量,外部无法访问。需修改为实例属性:

import random

class Patient:
    def __init__(self):
        self.x = random.randint(0,100)
        self.y = random.randint(0,100)

class Hospital:
    def __init__(self):
        self.x = random.randint(0,100)
        self.y = random.randint(0,100)

2. 实现距离计算与排序逻辑

使用欧几里得距离计算患者与医院的直线距离,给Patient类添加方法返回按距离排序的医院列表:

import math

class Patient:
    def __init__(self):
        self.x = random.randint(0,100)
        self.y = random.randint(0,100)
    
    def get_hospitals_sorted_by_distance(self, hospital_list):
        # 计算当前患者与每家医院的距离,存储为(医院, 距离)元组
        hospital_distances = []
        for hospital in hospital_list:
            distance = math.sqrt((self.x - hospital.x)**2 + (self.y - hospital.y)**2)
            hospital_distances.append((hospital, distance))
        
        # 按距离从小到大排序
        hospital_distances.sort(key=lambda item: item[1])
        # 返回排序后的医院列表(如需保留距离,可直接返回hospital_distances)
        return [item[0] for item in hospital_distances]

3. 使用示例

# 创建患者和医院列表
PatientList = [Patient() for _ in range(25)]
HospitalList = [Hospital() for _ in range(5)]

# 遍历患者,获取不同顺位的医院
for idx, patient in enumerate(PatientList):
    sorted_hospitals = patient.get_hospitals_sorted_by_distance(HospitalList)
    print(f"患者{idx+1}的匹配结果:")
    print(f"最近医院:坐标({sorted_hospitals[0].x}, {sorted_hospitals[0].y})")
    print(f"次近医院:坐标({sorted_hospitals[1].x}, {sorted_hospitals[1].y})")
    if len(sorted_hospitals) >=3:
        print(f"第三近医院:坐标({sorted_hospitals[2].x}, {sorted_hospitals[2].y})")
    print("---")

补充说明

  • 若需保留距离信息,可修改get_hospitals_sorted_by_distance方法,直接返回包含医院和距离的元组列表
  • 当前数据规模(25个患者+5家医院)下,直接计算完全足够;若数据量极大,可考虑KD-Tree等空间索引算法优化效率

内容的提问来源于stack exchange,提问作者user19678327

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.08.16 01:35:22