4

给定一个排序列表，例如[1.1, 2.2, 3.3]和一个边界值math.pi*2，返回最接近的任何给定值[0 - math.pi*2)

该函数应该返回值的索引，以便在f(1.2)返回0时f(2.1)返回1，并且f(6.0)应该环绕 atmath.pi*2和 return 0，在给定边界值的情况下，它更接近 1.1 而不是 3.3。只是为了完全明确，这个函数也应该在低端环绕，以便f(1.0, [5.0, 6.0], bound = math.pi*2)返回1。

用例是将弧度中的任意角度映射到列表中最近的现有有效角度。我已经在 python 中用 写过这种函数几次bisect，但代码总是会冒犯我的审美。边缘情况的高复杂性和数量似乎与函数的直观简单性不成比例。所以我想问是否有人能提出一个令人满意的实现，无论是在效率还是优雅方面。

4

3 回答 3

2

这是一种更优雅的方法。通过环绕数字线消除边缘情况：

from bisect import bisect_right

def circular_search(points, bound, value):
    ##
    ## normalize / sort input points to [0, bound)
    points = sorted(list(set([i % bound for i in points])))
    ##
    ## normalize search value to [0, bound)
    value %= bound
    ##
    ## wrap the circle left and right
    ext_points = [i-bound for i in points] + points + [i+bound for i in points]
    ##
    ## identify the "nearest not less than" point; no
    ## edge cases since points always exist above & below
    index = bisect_right(ext_points, value)
    ##
    ## choose the nearest point; will always be either the
    ## index found by bisection, or the next-lower index
    if abs(ext_points[index]-value) >= abs(ext_points[index-1]-value):
        index -= 1
    ##
    ## map index to [0, npoints)
    index %= len(points)
    ##
    ## done
    return points[index]

正如所写，除非输入像没有点一样不稳定，或者bound==0，否则有效。

于 2013-05-19T04:05:33.593 回答
1

使用bisect模块作为基础：

from bisect import bisect_left
import math

def f(value, sorted_list, bound=math.pi * 2):
    value %= bound
    index = bisect_left(sorted_list, value)
    if index == 0 or index == len(sorted_list):
        return min((abs(bound + sorted_list[0] - value), 0), (abs(sorted_list[-1] - value), len(sorted_list) - 1))[1]
    return min((index - 1, index), 
        key=lambda i: abs(sorted_list[i] - value) if i >= 0 else float('inf'))

演示：

>>> sorted_list = [1.1, 2.2, 3.3]
>>> f(1.2, sorted_list)
0
>>> f(2.1, sorted_list)
1
>>> f(6.0, sorted_list)
0
>>> f(5.0, sorted_list)
2
于 2013-05-18T09:45:12.920 回答
0

最简单的方法就是使用 min：

def angular_distance(theta_1, theta_2, mod=2*math.pi):
    difference = abs(theta_1 % mod - theta_2 % mod)
    return min(difference, mod - difference)

def nearest_angle(L, theta):
    return min(L, key=lambda theta_1: angular_distance(theta, theta_2))

In [11]: min(L, key=lambda theta: angular_distance(theta, 1))
Out[11]: 1.1

利用列表的排序性，您可以使用bisect模块：

from bisect import bisect_left

def nearest_angle_b(theta, sorted_list, mod=2*math.pi):
    i1 = bisect_left(sorted_list, theta % mod)
    if i1 == 0:
        i1, i2 = len(sorted_list) - 1, 0
    elif i1 == len(sorted_list):
        i1, i2 = i1 - 1, 0
    else:
        i2 = (i1 + 1) % len(sorted_list)
    return min((angular_distance(theta, L[i], mod), i, L[i])
                 for i in [i1, i2])

返回列表中最接近的距离、索引和角度。

于 2013-05-18T11:06:27.857 回答