包格式字符串中的自动重复标志

Question

在 php 中，unpack() 有“*”标志，表示“重复此格式直到输入结束”。例如，打印 97, 98, 99

$str = "abc";
$b = unpack("c*", $str);
print_r($b);

python 中有类似的东西吗？当然，我可以这样做

str = "abc"
print struct.unpack("b" * len(str), str)

，但我想知道是否有更好的方法。

Answer 1

在Python 3.4及更高版本中，您可以使用新函数 struct .iter_unpack。

struct.iter_unpack(fmt, buffer)

根据格式字符串fmt从缓冲区buffer中迭代解包。该函数返回一个迭代器，它将从缓冲区读取相同大小的块，直到其所有内容都被消耗。缓冲区的大小（以字节为单位）必须是格式所需大小的倍数，如 calcsize() 所反映的。

每次迭代都会生成一个由格式字符串指定的元组。

假设我们要使用重复格式字符串 '<2sc'（2 个字符后跟一个字符）解压数组 b'\x01\x02\x03'*3字符，重复直到完成）。

使用iter_unpack，您可以执行以下操作：

>>> import struct
>>> some_bytes = b'\x01\x02\x03'*3
>>> fmt = '<2sc'
>>> 
>>> tuple(struct.iter_unpack(fmt, some_bytes))
((b'\x01\x02', b'\x03'), (b'\x01\x02', b'\x03'), (b'\x01\x02', b'\x03'))

如果您想取消嵌套此结果，可以使用itertools.chain.from_iterable。

>>> from itertools import chain
>>> tuple(chain.from_iterable(struct.iter_unpack(fmt, some_bytes)))
(b'\x01\x02', b'\x03', b'\x01\x02', b'\x03', b'\x01\x02', b'\x03')

当然，您可以使用嵌套理解来完成同样的事情。

>>> tuple(x for subtuple in struct.iter_unpack(fmt, some_bytes) for x in subtuple)
(b'\x01\x02', b'\x03', b'\x01\x02', b'\x03', b'\x01\x02', b'\x03')

Answer 2

struct.unpack 中没有内置这样的工具，但可以定义这样的函数：

import struct

def unpack(fmt, astr):
    """
    Return struct.unpack(fmt, astr) with the optional single * in fmt replaced with
    the appropriate number, given the length of astr.
    """
    # http://stackoverflow.com/a/7867892/190597
    try:
        return struct.unpack(fmt, astr)
    except struct.error:
        flen = struct.calcsize(fmt.replace('*', ''))
        alen = len(astr)
        idx = fmt.find('*')
        before_char = fmt[idx-1]
        n = (alen-flen)/struct.calcsize(before_char)+1
        fmt = ''.join((fmt[:idx-1], str(n), before_char, fmt[idx+1:]))
        return struct.unpack(fmt, astr)

print(unpack('b*','abc'))
# (97, 98, 99)