找到最接近的三个数字的平均值-代码优化

def find_closest_average(nums):
    nums.sort()  # 对输入的数字列表进行排序

    closest_avg = float('inf')  # 初始化最接近的平均值为正无穷大
    closest_nums = []

    for i in range(len(nums) - 2):
        avg = (nums[i] + nums[i+1] + nums[i+2]) / 3  # 计算相邻三个数字的平均值

        if abs(avg - target_avg) < abs(closest_avg - target_avg):
            closest_avg = avg
            closest_nums = [nums[i], nums[i+1], nums[i+2]]

    return closest_avg, closest_nums

# 示例输入
nums = [1, 2, 3, 4, 5, 6, 7, 8, 9]
target_avg = sum(nums) / len(nums)  # 计算目标平均值

# 调用函数并输出结果
result_avg, result_nums = find_closest_average(nums)
print("最接近的三个数字的平均值：", result_avg)
print("最接近的三个数字：", result_nums)