python - 如何修改整数中的位？

Question

我有一个带有值7（0b00000111）的整数，我想用一个函数替换它13（0b00001101）。替换整数中的位的最佳算法是什么？

例如：

set_bits(somevalue, 3, 1) # What makes the 3rd bit to 1 in somevalue?

Answer 1

这些适用于任何大小的整数，甚至大于 32 位：

def set_bit(value, bit):
    return value | (1<<bit)

def clear_bit(value, bit):
    return value & ~(1<<bit)

如果你喜欢简短的东西，你可以使用：

>>> val = 0b111
>>> val |= (1<<3)
>>> '{:b}'.format(val)
'1111'
>>> val &=~ (1<<1)
'1101'

Answer 2

您只需要：

def set_bit(v, index, x):
  """Set the index:th bit of v to 1 if x is truthy, else to 0, and return the new value."""
  mask = 1 << index   # Compute mask, an integer with just bit 'index' set.
  v &= ~mask          # Clear the bit indicated by the mask (if x is False)
  if x:
    v |= mask         # If x was True, set the bit indicated by the mask.
  return v            # Return the result, we're done.

>>> set_bit(7, 3, 1)
15
>>> set_bit(set_bit(7, 1, 0), 3, 1)
13

请注意，位号 ( index) 从 0 开始，其中 0 是最低有效位。

另请注意，返回新值，无法像您显示的那样“就地”修改整数（至少我不这么认为）。

Answer 3

您可以使用按位操作。 http://wiki.python.org/moin/BitwiseOperators

如果要将给定位设置为 1，则可以在给定位置使用按位“或”和 1：

0b00000111 | 0b00001000 = 0b00001111

要将给定位设置为 0，您可以使用按位“和”

0b00001111 & 0b11111011 = 0b00001011

请注意，0b 前缀用于二进制数，0x 用于十六进制。

Answer 4

通过提供的示例，听起来您正在寻找以整数交换位。例如，在 7 中 (0b00000111)，如果您交换第 3 和第 1 位的位，您将获得 13 (0b00001101)。

我将以下内容作为函数签名swap_bits(val, i, j)

什么是最好的算法？好吧，下面的算法需要恒定的时间，O(1)。

def swap_bits(val, i, j):
    """
    Given an integer val, swap bits in positions i and j if they differ
    by flipping their values, i.e, select the bits to flip with a mask.
    Since v ^ 1 = 0 when v = 1 and 1 when v = 0, perform the flip using an XOR.
    """
    if not (val >> i) & 1 == (val >> j) & 1:
        mask = (1 << i) | (1 << j)
        val ^= mask
    return val

例子：

 >>> swap_bits(7, 3, 1)
 13

代码利用了一些小技巧，这里是 Sean Anderson 的一个很好的资源。我正在努力在此处提供 Python 中的代码片段。