功能追加/扩展

时间:2011-03-15 16:32:07

标签: python

Python中的方法appendextend本质上不起作用,它们修改被调用者并返回None

是否有其他方法可以执行这些方法的操作并将新列表作为返回值?

考虑这个例子:

def myfun(first, *args):
  for elem in [first].extend(args):
    print elem

显然,这不起作用。

有没有办法构建一个“就地”的新列表,而不是被迫编写以下内容?

def myfun(first, *args):
   all_args = list(first)
   all_args.extend(args)

   for elem in all_args:
     print elem

感谢。

2 个答案:

答案 0 :(得分:14)

>>> def append(lst, elem):
...     return lst + [elem]
... 
>>> append([1, 2, 3], 4)
[1, 2, 3, 4]
>>> def extend(lst1, lst2):
...   return lst1 + lst2
... 
>>> extend([1, 2], [3, 4])
[1, 2, 3, 4]

这就是你想要的吗?

您还可以定义自己的类型,除了更改之外,还会在这些操作上返回列表:

>>> class MyList(list):
...   def append(self, x):
...     super(MyList, self).append(x)
...     return self
...   def extend(self, lst):
...     super(MyList, self).extend(lst)
...     return self
... 
>>> l = MyList([1, 2, 3])
>>> l.append(4)
[1, 2, 3, 4]
>>> l.extend([5, 6, 7])
[1, 2, 3, 4, 5, 6, 7]
>>> l
[1, 2, 3, 4, 5, 6, 7]

答案 1 :(得分:6)

您可以将其重写为:

[first] + args