如何根据类型转换对列表项进行分组？

Question

针对您的需求，以下是对您现有代码进行调整后的解决方案：

data = [
    -1, 0,
    'a','b', 1, 2, 3,
    'c', 6,
    'd', 'e', .4, .5,
    'a', 'b', 4,
    'f', 'g',
]

start = list(map(type, data)).index(str)

wanted = {}
current_group = []
current_keys = []

for i in data[start:]:
    if type(i) == str:
        if current_keys:
            wanted['/'.join(current_keys)] = current_group
            current_keys = [i]
            current_group = []
        else:
            current_keys.append(i)
    else:
        current_group.append(i)

# 处理最后一个键值对
if current_keys:
    wanted['/'.join(current_keys)] = current_group

print(wanted)

此代码会输出您期望的结果：

{'a/b': [1, 2, 3, 4], 'c': [6], 'd/e': [0.4, 0.5], 'f/g': []}

hpaulj · Answer

您无需先找出 start。可以直接在执行剩余工作的循环中处理这一部分。

我的建议是，创建一个名为 current_key 的变量来追踪当前键，并在遇到新字符串时更新它，同时创建一个 current_values 列表来追踪该键对应的值。如果新字符串紧跟在一个字符串后面（即 current_values 为空），则修改当前键。如果不是这种情况，则视为新的键。若元素不是字符串，则将其追加到 current_values 列表中。

result = {}
current_key = []
current_vals = []
for ix, elem in enumerate(data):
    if isinstance(elem, str):
        if current_vals:  # 列表不为空
            key = '/'.join(current_key)
            if key not in result:
                result[key] = current_vals
            else:
                result[key].extend(current_vals)
            current_key = []
            current_vals = []

        current_key.append(elem)

    elif current_key:  # 只有在已遇到键的情况下才考虑值
        current_vals.append(elem)

# 最后，添加最后的键值对
key = '/'.join(current_key)
if key not in result:
    result[key] = current_vals
else:
    result[key].extend(current_vals)

在线试运行

Mehrdad Afshari · Answer

以下是经过详细注释的代码：

data = [
    -1, 0,
    'a','b', 1, 2, 3,
    'c', 6,
    'd', 'e', .4, .5,
    'a', 'b', 4,
    'f', 'g',
]
wanted = {}  # 初始化所需结果字典

i = 0  # 初始化数据列表的索引

# 跳过开头可能出现的整数值
while isinstance(data[i], int):  # 当元素为整数时
    i += 1  # 索引递增

while i < len(data):  # 当索引未超出范围时
    key = ''  # 初始化键值

    # 当元素为字符串（字符）时，构建键值
    while i < len(data) and isinstance(data[i], str):
        if len(key) > 0:  # 若键值已有字母，添加斜杠
            key += '/'
            
        key += data[i]  # 将字母添加至键值
        i += 1  # 索引递增

    # 若键值不在字典中，则添加它
    if key not in wanted:
        wanted[key] = []  # 初始化键值对应的新列表

    # 添加所有紧随字符串之后的数字至列表
    while i < len(data) and not isinstance(data[i], str):
        wanted[key].append(data[i])  # 将数字添加至列表
        i += 1  # 索引递增

print(wanted)

Adam Dymitruk · Answer

您可以使用 itertools.groupby函数配合一个测试当前项是否为字符串的key函数。为了跳过可能的非字符串开头项，先获取第一个分组，然后如果第一组项不是字符串，则再次获取下一个分组替换之。对于每一组项，如果它们是字符串，则将它们连接起来作为键；否则将这些项扩展到该键对应的列表中。如果最后一组项是字符串，则将其设置为默认的空列表：

from itertools import groupby, chain

output = {}
groups = groupby(data, lambda i: isinstance(i, str))
try:
    is_str, group = next(groups)
    if not is_str:
        is_str, group = next(groups)
except StopIteration:
    pass
else:
    for is_str, group in chain([(is_str, group)], groups):
        if is_str:
            key = '/'.join(group)
        else:
            output.setdefault(key, []).extend(group)
    if is_str:
        output.setdefault(key, [])

此时output将会是：

{'a/b': [1, 2, 3, 4], 'c': [6], 'd/e': [0.4, 0.5], 'f/g': []}

示例演示：https://ideone.com/rcOSzV

ekhumoro · Answer

blhsing答案的一个变体：

from itertools import groupby, chain

output = {}
groups = groupby(data, lambda i: isinstance(i, str))
nums = []
for is_str, group in groups:
    if is_str:
        nums = output.setdefault('/'.join(group), [])
    else:
        nums.extend(group)

在线尝试此代码！

或者：

output = {}
groups = groupby(data, lambda i: isinstance(i, str))
nums = None
for is_str, group in groups:
    if is_str:
        nums = output.setdefault('/'.join(group), [])
    elif nums is not None:
        nums.extend(group)

这两种变体都会生成预期的输出结果：

{'a/b': [1, 2, 3, 4], 'c': [6], 'd/e': [0.4, 0.5], 'f/g': []}