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

Question

[Курум]

Подскажите, пожалуйста.
Есть список:
p = [1,2,3,4,5,6,7,8,9]
Как из него сделать массив:

[[1,2,3],
 [4,5,6],
 [7,8,9]]

Comments

[Антон Рейтаровский]

Размер массива будет изменятся? Или только 9 элементов?

[x67]

список != массив в питоне, это очень важно помнить, особенно если вы привыкли к нормальному поведению массивов в любом языке программирования, поддерживающем срезы

[Курум]

Антон Рейтаровский: да, будет меняться

Answer 1

[aRegius]

[p[i:i+3] for i in range(0, 9, 3)]

Answer 2

[qlkvg]

Используйте numpy, если есть возможность

import numpy as np
arr = np.array([1,2,3,4,5,6,7,8,9])
arr.reshape(3,3)

Comments

[Курум]

а разделители меду числами можно добавить?

[qlkvg]

не совсем понимаю, что за разделители. приведите пример

[Курум]

Ответ получился вида:
[[1 2]
[3 4]
[5 6]
[7 8]]
А нужно, чтобы присутствовали запятые между числами и встроенными списками, т.е.:
[[1, 2],
[3, 4],
[5, 6],
[7, 8]]

[qlkvg]

эмм, ну сконвертируйте обратно в список через tolist(), или pprint для вывода используйте

Answer 3

[chupasaurus]

q = []
for i in range(len(p)//3):
    q.append(p[i*3:i*3+3])

if len(p) % 3 > 0:
    q.append(p[-1*(len(p)%3):])

Answer 4

[Roman K]

def chunked(s, n):
    current = 0
    while True:
        chunk = s[current : current+n]
        current += n
        if chunk:
            yield chunk
        else:
            break

In [6]: list(chunked(p, 3))
Out[6]: [[1, 2, 3], [4, 5, 6], [7, 8, 9]]

Работает с последовательностями любой (в т.ч. неизвестной) длины (напр. QuerySet Django), дробит на чанки любой длины.

Answer 5

[fdrwitch]

ну, как бы по классике:
n, m = 3, 3 #n - строки
mat = [[ (j+ i*2 + i ) for j in range(1, m+1)] for i in range(n)]
mat
[[1, 2, 3], [4, 5, 6], [7, 8, 9]]

Answer 6

[abcd0x00]

>>> p = [1, 2, 3, 4, 5, 6, 7, 8, 9]
>>> it = iter(p)
>>> list(map(list, zip(it, it, it)))
[[1, 2, 3], [4, 5, 6], [7, 8, 9]]
>>>

Answer 7

[Курум]

Можно чуть улучшить:

row = 3 # строк
col = int(len(p)/row) # столбцов

q = []
for i in range(len(p)//col):
    q.append(p[i*col:i*col+col])

if len(p) % col > 0:
    q.append(p[-1*(len(p)%col):])
print('q = ',q)