Как разделить список массивов на подмассивы?

Question

У меня есть список «а» со следующими значениями элемента.В моем коде я создал список:

a=[]
b=np.zeros(3)
c=[]
for i in range(0,4):
    b[0]=i+1
    b[1]=i+2
    b[2]=i+3
    c.append(deepcopy(b))
a.append(c)
c=[]
print(a)

Вывод:

[[array([1., 2., 3.]), array([2., 3., 4.]), array([3., 4., 5.]), array([4., 5., 6.])]]

Выше приведен пример, как я получаю в моих данных

Я пытался создать массив

b=np.array(a)

array([[[1., 2., 3.],
    [2., 3., 4.],
    [3., 4., 5.],
    [4., 5., 6.]]])
b.shape
(1,4,3)

Но я хочу сделать b формы (4,1,3), которая дает следующий вывод: чтобы при доступе к

b[0] gives [1,2,3]
b[1] gives [2,3,4]
b[2] gives [3,4,5]
b[3] gives [4,5,6]

Answer 1

Может быть, я что-то упустил, но вы сможете получить результат как:

b = np.array(a[0])

print(b[0]) # [1. 2. 3.]
print(b[1]) # [2. 3. 4.]
print(b[2]) # [3. 4. 5.]
print(b[3]) # [4. 5. 6.]

---

Для сохранения массива 3D:

np.array([a[0]]).reshape(4,1,3)

print(b[0]) #=> [[1. 2. 3.]]
print(b[1]) #=> [[2. 3. 4.]]
print(b[2]) #=> [[3. 4. 5.]]
print(b[3]) #=> [[4. 5. 6.]]

Answer 2

РЕДАКТИРОВАТЬ : используйте ответ @orli, его проще набрать!

Использование базового языка Python 3.

import numpy as np
from copy import deepcopy
a=[]
b=np.zeros(3)
c=[]
for i in range(0,4):
    b[0]=i+1
    b[1]=i+2
    b[2]=i+3
    c.append(deepcopy(b))
a.append(c)
res = []
for r in a:
    for c in r:
        rw = []
        for e in c.tolist():
            rw.append(e)
        res.append(rw)
print(res)

Выход:

[[1.0, 2.0, 3.0], [2.0, 3.0, 4.0], [3.0, 4.0, 5.0], [4.0, 5.0, 6.0]]

Answer 3

Для этого есть встроенная функция: b = np.vstack (a)

EDITED

После использования np.vstack (a)

b=b.reshape(4,3,1)

Это дает требуемый результат

b[0]- > [1,2,3]