将 numpy 数组添加到列表的特定索引

Question

我有一个 numpy 数组列表，想再添加两个数组作为列表的第一个 (0) 和最后一个 (-1) 索引。这些是我的列表和数组：

import numpy as np
first_arr = np.array([1,1,1])
last_arr = np.array([9,9,9])
middle_arr = [np.array ([4,4,4]), np.array ([6,6,6])]

然后，我想要一个 complete_arr 是：

complete_arr = [np.array ([1,1,1]), np.array ([4,4,4]), np.array ([6,6,6]), np.array ([9,9,9])]

我尝试了以下方法，但没有给我想要的结果：

insert_at = 0
middle_arr[insert_at:insert_at] = first_arr

如果有人帮助我在 python 中完成，我将不胜感激。

Answer 1

你可以使用*（解包）运算符：

first_arr = np.array([1,1,1])
last_arr = np.array([9,9,9])
middle_arr = [np.array ([4,4,4]), np.array ([6,6,6])]

complete_arr = [first_arr, *middle_arr, last_arr]

将 numpy 数组添加到列表的特定索引

adding numpy arrays to specific indices of a list

python

numpy

list

insert

append