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

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 从缓冲区缓冲区中迭代解包。此函数返回一个迭代器，它将从缓冲区读取相同大小的块，直到其所有内容都已被消耗。缓冲区的字节大小必须是格式所需大小的倍数，如 calcsize() 所反映的那样。

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

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

使用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)