您如何将列表中的多个数字相加并合并？ （Python）

Question

所以，假设我有以下列表和变量 i：

data = [ [1, 2, 3, 1, 1], [1, 2, 3, 0, 0], [1, 2, 3, 2, 1] ]
i = 3

我想创建一个新列表，该列表将汇总并合并每个子列表中第 i 个元素向上的数字以生成以下内容：

new_data = [ [1, 2, 5], [1, 2, 3], [1, 2, 6] ]

我从每个子列表中向上求和第三个元素并将这些元素合并在一起。我想问一下，通过使用 Python 中的标准库，我可以为 i 的任意（整数）值实现这一点。我有以下想法，但我不确定如何将其放入真正的 Python 代码中：

（伪代码）

new_data = []
for sublist in data:
      new_data.append(elements before the ith element)
      new_data.append(ith element onwards summed together)

Answer 1

您可以按索引对内部列表进行切片，然后将其余部分与单行列相加：

>>> new_data = [row[:i-1] + [sum(row[i-1:])] for row in data]
>>> new_data
[[1, 2, 5], [1, 2, 5], [1, 2, 5]]

Answer 2

您可以在taras' answer 中找到一个不错的“pythonic”单行：

new_data = [row[:i-1] + [sum(row[i-1:])] for row in data].

从伪代码到正确的python代码

我的回答重点是帮助您将伪代码转换为 python 代码。

伪代码：

new_data = []
for sublist in data:
      new_data.append(elements before the ith element)
      new_data.append(ith element onwards summed together)

您的伪代码的第一个问题是您要区分列表 data 及其子列表 sublist，但您没有区分列表 new_data 及其子列表。让我在循环中添加一个变量new_sublist：

new_data = []
for sublist in data:
      new_sublist = []
      new_sublist.append(elements before the ith element)
      new_sublist.append(ith element onwards summed together)
      new_data.append(new_sublist)

您的伪代码的第二个问题：您在循环的每次迭代中对.append 进行两次调用。但是，这两个调用并不相似：第一个调用应该附加几个元素，而第二个调用应该附加一个元素。 Python 对这两种操作进行了区分；如果您想一次添加多个元素，请使用.extend 而不是.append。代码变为：

new_data = []
for sublist in data:
      new_sublist = []
      new_sublist.extend([elements before the ith element])
      new_sublist.append(ith element onwards summed together)
      new_data.append(new_sublist)

最后，我们可以将您的伪代码转换为正确的 python 代码，使用列表切片来获取第 i 个元素之前的元素，并使用内置函数 sum 以及列表切片来对第 i 个元素进行求和：

new_data = []
for sublist in data:
      new_sublist = []
      new_sublist.extend(sublist[:i])
      new_sublist.append(sum(sublist[i:]))
      new_data.append(new_sublist)

请注意，在 python 中，循环遍历一个列表以构造一个新列表是一种非常常见的操作，我们使用紧凑而优雅的列表解析语法来代替多行 for-loop：

new_data = [sublist[:i] + [sum(sublist[i:])] for sublist in data]

相关文档

• list.extend and list.append;
• list slices;
• builtin function sum;
• list comprehensions。

Answer 3

你可以这样做：

def merger(data, pos):
    pos -= 1 # 0 based, your i above is 1 based

    # collection list for result
    result = []
    # go over each given inner part
    for inner in data:
        # add it up to the correct position
        result.append(inner[:pos])
        # extend it by the sum of the remainder values
        result[-1].append(sum(inner[pos:]))

    return result

data = [ [1, 2, 3, 1, 1], [1, 2, 3, 1, 1], [1, 2, 3, 1, 1] ]
i = 3

print(merger(data,i)) 

输出：

[[1, 2, 5], [1, 2, 5], [1, 2, 5]]

或者简而言之：

def merger(data, pos):
    return [[inner[: pos - 1] + [sum(inner[pos - 1 :])] for inner in data]]

Answer 4

data = [ [1, 2, 3, 1, 1], [1, 2, 3, 1, 1], [1, 2, 3, 1, 1] ]
i = 3
new_data = [sublist[:i-1] + [sum(sublist[i-1:])] for sublist in data]
print(new_data)
>>> [[1, 2, 5], [1, 2, 5], [1, 2, 5]]