Python if-else代码样式，用于减少浮点取整的代码

Question

有没有更短，更清晰的代码样式可以解决此问题？我正在尝试将一些float值分类为区域间文件夹。

def classify(value):   
    if value < -0.85 and value >= -0.95:
        ts_folder = r'\-0.9'
    elif value < -0.75 and value >= -0.85:
        ts_folder = r'\-0.8'
    elif value < -0.65 and value >= -0.75:
        ts_folder = r'\-0.7'    
    elif value < -0.55 and value >= -0.65:
        ts_folder = r'\-0.6'   
    elif value < -0.45 and value >= -0.55:
        ts_folder = r'\-0.5'  
    elif value < -0.35 and value >= -0.45:
        ts_folder = r'\-0.4'
    elif value < -0.25 and value >= -0.35:
        ts_folder = r'\-0.3'
    elif value < -0.15 and value >= -0.25:
        ts_folder = r'\-0.2'
    elif value < -0.05 and value >= -0.15:
        ts_folder = r'\-0.1'
    elif value < 0.05 and value >= -0.05:
        ts_folder = r'\0.0'
    elif value < 0.15 and value >= 0.05:
        ts_folder = r'\0.1'
    elif value < 0.25 and value >= 0.15:
        ts_folder = r'\0.2'
    elif value < 0.35 and value >= 0.25:
        ts_folder = r'\0.3'
    elif value < 0.45 and value >= 0.35:
        ts_folder = r'\0.4'
    elif value < 0.55 and value >= 0.45:
        ts_folder = r'\0.5'
    elif value < 0.65 and value >= 0.55:
        ts_folder = r'\0.6'
    elif value < 0.75 and value >= 0.65:
        ts_folder = r'\0.7'  
    elif value < 0.85 and value >= 0.75:
        ts_folder = r'\0.8'
    elif value < 0.95 and value >= 0.85:
        ts_folder = r'\0.9'

    return ts_folder

Answer 1

具体解决方案

没有真正的通用解决方案，但是您可以使用以下表达式。

ts_folder = r'\{:.1f}'.format(round(value, 1))

一般解决方案

如果您实际上需要某种概括，请注意任何非线性模式都会引起麻烦。虽然，有一种方法可以缩短代码。

def classify(key, intervals):
    for lo, hi, value in intervals:
        if lo <= key < hi:
            return value
    else:
        ... # return a default value or None

# A list of tuples (lo, hi, key) which associates any value in the lo to hi interval to key
intervals = [
    (value / 10 - 0.05, value / 10 + 0.05, r'\{:.1f}'.format(value / 10))
    for value in range(-9, 10)
]

value = -0.73

ts_folder = classify(value, intervals) # r'\-0.7'

请注意，上述内容仍然不能完全避免某些float舍入错误。您可以通过手动键入intervals列表而不是使用理解来增加精度。

连续间隔

如果您的数据中的间隔是连续的，那么它们之间就没有间隙，如您的示例所示，那么我们可以使用一些优化方法。即，我们只能在列表中存储每个间隔的上限。然后，通过对它们进行排序，我们可以bisect进行有效的查找。

import bisect

def value_from_hi(hi):
    return r'\{:.1f}'.format(hi - 0.05)

def classify(key, boundaries):
    i = bisect.bisect_right(boundaries, key)
    if i < len(boundaries):
        return value_from_hi(boundaries[i])
    else:
        ... # return some default value

# Sorted upper bounds
boundaries = [-0.85, -0.75, -0.65, -0.55, -0.45, -0.35, -0.25, -0.15, -0.05,
              0.05, 0.15, 0.25, 0.35, 0.45, 0.55, 0.65, 0.75, 0.85, 0.95]

ts_folder = classify(-0.32, boundaries) # r'\-0.3'

重要说明：选择使用更高的界限bisect_right是由于您的示例中没有更高的界限。如果排除了下限，则我们必须将其与一起使用bisect_left。

另请注意，您可能希望以某些特殊方式处理[-0.95、0.95 []范围以外的数字，并注意将其保留为bisect。

Answer 2

bisect模块将执行正确的查找，以便从断点列表中找到正确的bin。实际上，文档中的示例就是这样的情况：

bisect（）函数通常可用于对数字数据进行分类。本示例使用bisect（）根据一组有序的数字断点来查找考试总计（例如）的字母等级：85及以上为'A'，75..84为'B'，依此类推。

>>> grades = "FEDCBA"
>>> breakpoints = [30, 44, 66, 75, 85]
>>> from bisect import bisect
>>> def grade(total):
...           return grades[bisect(breakpoints, total)]
>>> grade(66)
'C'
>>> map(grade, [33, 99, 77, 44, 12, 88])
['E', 'A', 'B', 'D', 'F', 'A']

您需要一个字符串列表，而不是用于值查找的字符串，该字符串列表用于每个值范围的确切文件夹名称。例如：

breakpoints = [-0.85, -0.75, -0.65]
folders = [r'\-0.9', r'\-0.8', r'\-0.7']
foldername = folders[bisect(breakpoints, -0.72)]

当然，如果您可以自动化此表生成的一部分（使用round()或类似方法），则应该这样做。

Answer 3

带有这样的代码块的第一条规则之一是，始终使比较方向相同。所以代替

    elif value < -0.75 and value >= -0.85:

写

    elif -0.85 <= value and value < -0.75:

在这一点上，您可以观察到python允许链接比较，因此您可以编写：

    elif -0.85 <= value < -0.75:

这本身就是一种改进。另外，您可以观察到这是比较的有序列表，因此，如果添加初始比较，则只需编写

    if value < -0.95:        ts_folder = ''
    elif value < -0.85:      ts_folder = r'\-0.9'
    elif value < -0.75:      ts_folder = r'\-0.8'
    elif value < -0.65:      ts_folder = r'\-0.7'    
    elif value < -0.55:      ts_folder = r'\-0.6'   
    elif value < -0.45:      ts_folder = r'\-0.5'  
    elif value < -0.35:      ts_folder = r'\-0.4'
    elif value < -0.25:      ts_folder = r'\-0.3'
    elif value < -0.15:      ts_folder = r'\-0.2'
    elif value < -0.05:      ts_folder = r'\-0.1'
    elif value < 0.05:       ts_folder = r'\0.0'
    elif value < 0.15:       ts_folder = r'\0.1'
    elif value < 0.25:       ts_folder = r'\0.2'
    elif value < 0.35:       ts_folder = r'\0.3'
    elif value < 0.45:       ts_folder = r'\0.4'
    elif value < 0.55:       ts_folder = r'\0.5'
    elif value < 0.65:       ts_folder = r'\0.6'
    elif value < 0.75:       ts_folder = r'\0.7'  
    elif value < 0.85:       ts_folder = r'\0.8'
    elif value < 0.95:       ts_folder = r'\0.9'
    else:                    ts_folder = ''

那仍然很长，但是a）可读性更高；b）它有明确的代码要处理value < -0.95 or 0.95 <= value

Answer 4

您可以使用round()内置的：

ts_folder = "\\" + str(round(value + 1e-16, 1)) # To round values like .05 to .1, not .0
if ts_folder == r"\-0.0": ts_folder = r"\0.0" 

更多关于 round()

Answer 5

所有答案都围绕四舍五入，这在这种情况下似乎很好，但仅出于论证的目的，我还想指出一个很酷的python字典用法，它通常被描述为其他语言开关的替代方法进而允许使用任意值。

ranges = {
    (-0.85, -0.95): r'\-0.9',
    (-0.75, -0.85): r'\-0.8',
    (-0.65, -0.75): r'\-0.7',
    (-0.55, -0.65): r'\-0.6'
    ...
}

def classify (value):
    for (ceiling, floor), rounded_value in ranges.items():
        if floor <= value < ceiling:
            return rounded_value

输出：

>>> classify(-0.78)
\-0.8

Answer 6

实际上，在Python 3 .85中将四舍五入.8。根据问题.85应该舍入.9。

您可以尝试以下方法：

round2 = lambda x, y=None: round(x+1e-15, y)
ts_folder = r'\{}'.format(str(round2(value, 1)))

输出：

>>> round2(.85, 1)
0.9
>>> round2(-.85, 1)
-0.8