У меня есть вложенный список, но я хочу, чтобы добавить вложенный список внутри вложенного списка

Question

У меня есть список, как:

[[21, 32, 32], [23, 34, 32], [32, 34, 57]] 

Я хотел бы преобразовать каждый номер в своем собственном списке, как

[[[21], [32], [32]], [[23], [34], [32]], [[32], [34], [57]]] 

Как мне это сделать?

Спасибо!

Answer 1

Используйте список понимание:

>>> lst = [[21, 32, 32], [23, 34, 32], [32, 34, 57]] 
>>> new_lst = [[[i] for i in sub_lst] for sub_lst in lst] 
>>> new_lst 
[[[21], [32], [32]], [[23], [34], [32]], [[32], [34], [57]]] 

---

Вы можете также использовать numpy, просто добавив дополнительную ось с помощью np.newaxis в то время как нарезка ваш массив:

>>> import numpy as np 
>>> l = np.array([[21, 32, 32], [23, 34, 32], [32, 34, 57]]) 
>>> l[:, :, np.newaxis] # or l[:, :, None] 
array([[[21], 
     [32], 
     [32]], 

     [[23], 
     [34], 
     [32]], 

     [[32], 
     [34], 
     [57]]]) 

Answer 2

Здесь метод с использованием numpy.reshape():

import numpy as np 
arr = np.array([[21, 32, 32], [23, 34, 32], [32, 34, 57]]) 

arr.reshape(3,3,1) 

# array([[[21], 
#   [32], 
#   [32]], 

#  [[23], 
#   [34], 
#   [32]], 

#  [[32], 
#   [34], 
#   [57]]]) 

Answer 3

Метод с использованием списка понимание:

a = [[21, 32, 32], [23, 34, 32], [32, 34, 57]] 

print [[[element] for element in each] for each in a] 

Выход:

[[[21], [32], [32]], [[23], [34], [32]], [[32], [34], [57]]]