这种while循环写法是否规范？如何优化求≥目标值的最小2的幂函数

我想写一个while循环，找出大于等于给定目标值的最近2的幂并返回。当前的power_of_two函数大部分情况都能正常工作，但当目标值为1时会出错。试过调整count的值，但没找到更好的办法，希望能不使用if语句优化这个函数。

原实现代码：

def power_of_two(target):
    """
    -------------------------------------------------------
    Determines the nearest power of 2 greater than or equal to
    a given target.
    Use: power = power_of_two(target)
    -------------------------------------------------------
    Parameters:
        target - value to find nearest power of 2 (int >= 0)
    Returns:
        power - first power of 2 >= target (int)
    -------------------------------------------------------
    """
    x = 0
    count = 0
    while target > x:
        count += 1
        x = 2**count
    if target == 1:
        count = 1
    else:
        count = 2**count
    return count

问题分析 #

原代码的核心问题是初始值设置不当：x从0开始，count从0开始，当target=1时，循环会执行一次（因为1>0成立），此时count变成1，x变成2^1=2，循环结束后如果直接返回2**count会得到2，这和预期的1不符，所以不得不加if语句修正。

优化后的代码（无if语句） #

def power_of_two(target):
    """
    -------------------------------------------------------
    Determines the nearest power of 2 greater than or equal to
    a given target.
    Use: power = power_of_two(target)
    -------------------------------------------------------
    Parameters:
        target - value to find nearest power of 2 (int >= 0)
    Returns:
        power - first power of 2 >= target (int)
    -------------------------------------------------------
    """
    power = 1
    while power < target:
        power *= 2
    return power

优化思路 #

1. 调整初始值：直接把power初始化为1（也就是2^0，最小的正2的幂），避免从0开始的无效循环。
2. 修改循环条件：用power < target代替原有的target > x，这样当target<=1时，循环直接不执行，直接返回1，完美解决target=1的问题。
3. 简化逻辑：每次循环直接将power乘以2，不需要额外维护count变量，代码更简洁直观。

测试验证：

• target=1 → 返回1（正确）
• target=2 → 返回2（正确）
• target=3 → 返回4（正确）
• target=0 → 返回1（符合“大于等于0的最近2的幂”要求）

内容的提问来源于stack exchange，提问作者user3542203