python：将元素组合在一起

Question

我目前正在使用itertools创建并返回一个列表，其元素是包含原始列表的相等元素的连续运行的列表。

import itertools
it = [1, 1, 5, 5, 5, 'test', 'test', 5]

new = len(it)
for a in range(new):
  return [list(k) for a, k in itertools.groupby(it)] 

对于上面的例子，结果是：

[[1, 1], [5, 5, 5], ['test', 'test'], [5]]

我可以在不使用itertools的情况下实现这一目标吗？

Answer 1

您可以通过将列表与自身压缩但使用float('nan')的填充来配对相邻项目，因为它不能等于任何对象，然后遍历压缩对以将项目附加到输出列表的最后子列表，并添加当相邻项目不同时，新的子列表：

output = []
for a, b in zip([float('nan')] + it, it):
    if a != b:
        output.append([])
    output[-1].append(b)

output成为：

[[1, 1], [5, 5, 5], ['test', 'test'], [5]]

Answer 2

说实话，一个简单的for循环可以使这个工作，你甚至不必导入itertools。

最简单的方法是使用这个：

it = [1, 1, 5, 5, 5, 'test', 'test', 5]
result = []
for (i, x) in enumerate(it):
  if i < 1 or type(x) != type(it[i - 1]) or x != it[i - 1]:
    result.append([x])
  else:
    result[-1].append(x)
print(result)

或者，在功能形式：

def type_chunk(it):
  result = []
  for (i, x) in enumerate(it):
    if i < 1 or type(x) != type(it[i - 1]) or x != it[i - 1]:
      result.append([x])
    else:
      result[-1].append(x)
  return result

然后你会使用这样的函数：

print(type_chunk([1, 1, 5, 5, 5, 'test', 'test', 5]))

你甚至可以跳过类型检查，只查找相同的值：

def type_chunk(it):
  result = []
  for (i, x) in enumerate(it):
    if i < 1 or x != it[i - 1]:
      result.append([x])
    else:
      result[-1].append(x)
  return result

祝好运。

Answer 3

您可以查看itertools中的函数，了解它们是如何进行的。

这是一种清楚地显示逻辑的方法（可以进一步减少）：

def i_am_itertool():
    it = [1, 1, 5, 5, 5, 'test', 'test', 5]
    ret = []

    temp = []
    last = it[0]
    for e in it:
        if e == last:
            temp.append(e)
        else:
            ret.append(temp)  # Add previous group
            temp = [e]  # Start next group
            last = e
    ret.append(temp)  # Add final group
    return ret

print(i_am_itertool())

输出：

 [[1, 1], [5, 5, 5], ['test', 'test'], [5]]