代码之家 › 专栏 › 技术社区 › jxie0755

sum()如何用于合并列表

python

jxie0755 · 技术社区 · 7 年前

def tree(label, branches=[]):
    return [label] + list(branches)

def label(tree):
    return tree[0]

def branches(tree):
    return tree[1:]

有一个函数可以将树的所有节点提取到列表中:

def all_nodes(tree):
    return [label(tree)] + sum([all_nodes(b) for b in branches(tree)], [])

T = tree(1, [tree(2, [tree(4), tree(5)]), tree(3, [tree(6), tree(7)])])
print(all_nodes(T))
# >>> [1, 2, 4, 5, 3, 6, 7]

你可以看出这很有效,但我搞不懂 sum() 在这里使用。

我知道一个列表可以添加到另一个列表:

print([1] + [2]) # >>> [1, 2]

求和() :

a, b = [1], [2]
print(sum(a, b))
# >>> TypeError: can only concatenate list (not "int") to list
print(sum([a, b]))
# >>> TypeError: unsupported operand type(s) for +: 'int' and 'list

在 tree 函数,如何 求和()

2 回复 | 直到 7 年前

Prune 7 年前

sum 对一系列元素进行操作,例如 sum([1, 2, 3]) (生产 6 )或者 sum([ [1], [2] ], []) (生产 [1, 2] ). 有一个可选的第二个参数 start 价值。例如, sum([1, 2, 3], 10) 从10开始计算,假设 16 开始 默认为 0 :如果要对非数字对象求和,则必须提供 开始 价值。

当你给它的时候 sum(a, b) ,列表 a 总和 did是(正确地)遍历该列表的项,将它们添加到你提供的价值。逻辑是这样的:

result = b
for element in a:
    result = result + element

因此,你要做的第一件事就是 result = [2] + 1 . 记住,第一个论点是序列

sum([a], b)

产生 [2, 1] b 是起始值。

Olivier Melançon iacob 7 年前

内建方法 sum +

默认情况下,起始值为 0 ,含义 sum([[1], [2]]) 相当于 0 + [1] + [2] 它引发了 TypeError .

要连接希望初始值为 [] ,空列表。那么, sum([[1], [2], [3]], []) [] + [1] + [2] + [3] 如所愿。

性能

不建议使用列表已创建。相反,您希望使用遍历所有列表并将项目附加到新列表的解决方案。

def concat_lists(lists):
    new_list = []
    for l in lists:
        new_list.extend(l)

或者使用 itertools .

from itertools import chain

new_list = list(chain(*lists))

推荐文章

Google User · Django管理员在`list_display中未显示`creation_date`字段`

1 年前

user29747013 · 如何创建一个新的数据框架,其中包含原始数据框架中列的聚合列?

1 年前

ÎÎÎ½Î· ÎÎ®Î¹Î½Î¿Ï · Python lxml.html语法错误:使用lxml find时XPATH的谓词无效

1 年前

user29715306 · from_users=和chats=电视节目中的差异

1 年前

Redshoe · 当执行numpy.genfromtxt()时,python是否会读取文件的所有行?

1 年前

RASEL MAHMUD · 为什么以及如何在is_even()函数内的IF条件中递归X变量在满足0后递增?[副本]

1 年前

prayner · 更新嵌套字典包含列表中的项

1 年前

Bringo Jr · 我可以在O(n)中解决这个问题吗?

1 年前

Dave · 如何在for循环中修改列表值

1 年前

Shukurullox Komiljonov · 从记录中获得相互和解。使用SQL

1 年前