我有一个存储在列表中的数组,并希望将该列表拆分成子列表。这是我的列表:
all_data=[np.array([[1., 2.]]), np.array([[7., 7.], [4., 1.]]),\
np.array([[-1., 4.], [1., 9.]]), np.array([[3., 0.]]),\
np.array([[0., -2.]]), np.array([[6., 1.], [3., 5.]])]
它有6个数组,我想把它分成两个子列表。第一个子列表包括前三个数组,第二个子列表包括后三个数组。结果如下:
spl_data=[[np.array([[1., 2.]]), np.array([[7., 7.], [4., 1.]]),\
np.array([[-1., 4.], [1., 9.]])],\
[np.array([[3., 0.]]),\
np.array([[0., -2.]]), np.array([[6., 1.], [3., 5.]])]]
我尝试了以下函数:
def chunks(lst, n):
for i in range(0, len(lst), n):
yield lst[i:i + n]
接着我尝试了以下方法来达到想要的结果,但是并没有成功:
n=2
spl_data=list(chunks(all_data, n))
我非常感激您提前的帮助。
a,b = all_data[:3],all_data[3:]
来拆分列表,这与numpy
无关(或者将a,b = ...
替换为spl_data = ...
)。 - Pierre D