在列表中的特定索引处插入元素并返回更新后的列表

Question

我有这个：

>>> a = [1, 2, 4]
>>> print a
[1, 2, 4]

>>> print a.insert(2, 3)
None

>>> print a
[1, 2, 3, 4]

>>> b = a.insert(3, 6)
>>> print b
None

>>> print a
[1, 2, 3, 6, 4]

有没有一种方法可以获得更新的列表作为结果，而不是更新原始列表？

Answer 1

l.insert(index, obj) 实际上并没有返回任何东西。它只是更新列表。

正如 ATO 所说，您可以使用b = a[:index] + [obj] + a[index:]。 但是，另一种方式是：

a = [1, 2, 4]
b = a[:]
b.insert(2, 3)

Answer 2

最高效的方法

您也可以使用 slice 索引 在列表中插入元素。例如：

>>> a = [1, 2, 4]
>>> insert_at = 2  # Index at which you want to insert item

>>> b = a[:]   # Created copy of list "a" as "b".
               # Skip this step if you are ok with modifying the original list

>>> b[insert_at:insert_at] = [3]  # Insert "3" within "b"
>>> b
[1, 2, 3, 4]

对于在给定索引处同时插入多个元素，您只需使用要插入的多个元素的list。例如：

>>> a = [1, 2, 4]
>>> insert_at = 2   # Index starting from which multiple elements will be inserted

# List of elements that you want to insert together at "index_at" (above) position
>>> insert_elements = [3, 5, 6]

>>> a[insert_at:insert_at] = insert_elements
>>> a   # [3, 5, 6] are inserted together in `a` starting at index "2"
[1, 2, 3, 5, 6, 4]

了解更多切片索引，可以参考：Understanding slice notation。

注意：在 Python 3.x 中，slice indexing 和 list.index(...) 之间的性能差异显着减小，两者几乎相等.但是，在 Python 2.x 中，这种差异非常明显。我在这个答案后面分享了性能比较。

---

使用列表理解的替代方案 （但在性能方面非常慢）：

作为替代方案，也可以使用 list comprehension 和 enumerate 来实现。 （但请不要这样做。这只是为了说明）：

>>> a = [1, 2, 4]
>>> insert_at = 2

>>> b = [y for i, x in enumerate(a) for y in ((3, x) if i == insert_at else (x, ))]
>>> b
[1, 2, 3, 4]

---

所有解决方案的性能比较

这是timeit 将所有答案与 Python 3.9.1 和 Python 2.7.16 上的 1000 个元素列表的比较。答案按两个 Python 版本的性能顺序列出。

Python 3.9.1

1. My answer 使用切片插入 - 最快（每个循环 2.25 微秒）

   python3 -m timeit -s "a = list(range(1000))" "b = a[:]; b[500:500] = [3]"
   100000 loops, best of 5: 2.25 µsec per loop
   

2. Rushy Panchal's answer 使用list.insert(...) 获得最多票数 - 第二（每个循环 2.33 微秒）

   python3 -m timeit -s "a = list(range(1000))" "b = a[:]; b.insert(500, 3)"
   100000 loops, best of 5: 2.33 µsec per loop
   

3. ATOzTOA's accepted answer 基于切片列表的合并 - 第三个（每个循环 5.01 微秒）

   python3 -m timeit -s "a = list(range(1000))" "b = a[:500] + [3] + a[500:]"
   50000 loops, best of 5: 5.01 µsec per loop
   

4. My answer 与 List Comprehension 和 enumerate - 第四 （非常慢，每个循环 135 微秒）

   python3 -m timeit -s "a = list(range(1000))" "[y for i, x in enumerate(a) for y in ((3, x) if i == 500 else (x, )) ]"
   2000 loops, best of 5: 135 µsec per loop
   

Python 2.7.16

1. My answer 使用切片插入 - 最快（每个循环 2.09 微秒）

   python -m timeit -s "a = list(range(1000))" "b = a[:]; b[500:500] = [3]"
   100000 loops, best of 3: 2.09 µsec per loop
   

2. Rushy Panchal's answer 使用list.insert(...) 获得最多票数-第二（每个循环2.36 µsec）

   python -m timeit -s "a = list(range(1000))" "b = a[:]; b.insert(500, 3)"
   100000 loops, best of 3: 2.36 µsec per loop
   

3. ATOzTOA's accepted answer 基于切片列表的合并 - 第三个（每个循环 4.44 微秒）

   python -m timeit -s "a = list(range(1000))" "b = a[:500] + [3] + a[500:]"
   100000 loops, best of 3: 4.44 µsec per loop
   

4. My answer 与 List Comprehension 和 enumerate - 第四（非常慢，每个循环 103 微秒）

   python -m timeit -s "a = list(range(1000))" "[y for i, x in enumerate(a) for y in ((3, x) if i == 500 else (x, )) ]"
   10000 loops, best of 3: 103 µsec per loop
   

Answer 3

我得到的最短的：b = a[:2] + [3] + a[2:]

>>>
>>> a = [1, 2, 4]
>>> print a
[1, 2, 4]
>>> b = a[:2] + [3] + a[2:]
>>> print a
[1, 2, 4]
>>> print b
[1, 2, 3, 4]

Answer 4

最干净的方法是复制列表，然后将对象插入到副本中。在 Python 3 上，这可以通过 list.copy 完成：

new = old.copy()
new.insert(index, value)

在 Python 2 上，可以通过 new = old[:] 复制列表（这也适用于 Python 3）。

就性能而言，与其他提议的方法没有区别：

$ python --version
Python 3.8.1
$ python -m timeit -s "a = list(range(1000))" "b = a.copy(); b.insert(500, 3)"
100000 loops, best of 5: 2.84 µsec per loop
$ python -m timeit -s "a = list(range(1000))" "b = a.copy(); b[500:500] = (3,)"
100000 loops, best of 5: 2.76 µsec per loop

Answer 5

这是添加单个项目的方法，特定索引中的单个项目将列表与另一个列表连接

>>> expences = [2200, 2350, 2600, 2130, 2190]
>>> expences.append(1980)
>>> expences
[2200, 2350, 2600, 2130, 2190, 1980]

>>> expences.insert(1, 1200)

>>> expences
[2200, 1200, 2350, 2600, 2130, 2190, 1980]

>>> newElm = [2550, 2123, 2430]
>>> expences.extend(newElm)
>>> expences
[2200, 1200, 2350, 2600, 2130, 2190, 1980, 2550, 2123, 2430]
>>> 

Answer 6

使用Python list insert() method。用法：

#语法

insert() 方法的语法 -

list.insert(index, obj)

#参数

• index - 这是需要插入对象 obj 的索引。
• obj - 这是要插入给定列表的对象。

#返回值 此方法不返回任何值，但它在给定索引处插入给定元素。

例子：

a = [1,2,4,5]

a.insert(2,3)

print(a)

返回[1, 2, 3, 4, 5]