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

Question

[numb7]

Как создать многомерный список, чтобы получилось n списков? Мое решение позволяет получить n - элементов.

Мое решение:

n = 3
m = 2
a = [0] * n
for i in range(n):
    a[i] = [0] * m
print(a)

Текущий ответ: [[0, 0], [0, 0], [0, 0]].
Ответ, который должен получиться: [[[0, 0], [0, 0]], [[0, 0], [0, 0]]].

Answer 1

[Vindicar]

Т.е. тебе нужно N уровней вложенности, но что насчёт размера? Он всегда 2, по каждой оси - т.е. 2х2 для двухмерного списка, 2х2х2 для трёхмерного и т.д.?
Если да, то могу подсказать. Первое измерение ты генерируешь правильно через [0] * k (где k - размер списка по последней оси).
А вот дальше можно легко добавить ещё одно измерение такой конструкцией:
a = [a.copy() for i in range(k)]
Так как мы делаем copy(), каждый вложенный список будет отдельным объектом, а не ссылками на один и тот же объект.
Это добавление одного измерения. Как добавить столько, сколько нужно, догадаешься сам, я думаю.

Comments

[numb7]

1. Да, n - уровней вложенности
2. Да, всегда 2x2 (если я правильно тебя понял)
для n = 2 ответ будет
[[0, 0], [0, 0]]

[Vindicar]

numb7, тогда мой ответ должен сработать. Просто прими k=2 и наращивай уровни вложенности списка в цикле.

[numb7]

Vindicar, не совсем, твой вариант выводит: [[0,0],[0,0]...n]

[Vindicar]

numb7,

Это добавление одного измерения. Как добавить столько, сколько нужно, догадаешься сам, я думаю.

наращивай уровни вложенности списка в цикле

Ну и не путай k (длина списка по оси) и n (размерность).

Answer 2

[PavelMos]

Нужен 3-мерный список, то есть массив вложенный в массив вложенный в массив ?
Тогда нужно 3 прохода- через for, через генератор списков, через генератор вложенных списков. Необязательно три за раз, можно 1+2 или 2+1

s=[ [ [0 for x in range (0,2)] for y in range (0,2) ] for z in range (0,2)]

Out[20]: [[[0, 0], [0, 0]], [[0, 0], [0, 0]]]

Comments

[numb7]

Если n = 3, то получится 3-мерный список, если n = 2, то будет двумерный список

[PavelMos]

Это более сложно тк здесь уже будет разный алгоритм в случае разной вложенности. Так как все циклы со всеми переменными будут в одной области видимости, у всех должны быть разные имена переменных i,j.... и их надо задать заранее.

Как вариант - сделать для максимальной вложенности при переходе на каждый уровень делать проверку n.
Если большАя вложенность не нужна, эта часть цикла пропускается.
Или - делать через eval, т.е. сначала составлять выражение, например, вложенный генератор списков, и затем выполнять его и сохранять результат