Нарезка списков питонов

1

Если у меня есть список say 'n' элементов (каждый элемент представляет собой один байт), который представляет прямоугольную матрицу 2d, как я могу разбить это на прямоугольники, например w * h, начиная с первого элемента списка, просто используя стандартные функции python

например

l =  
 [ 1,2,3,4,5,6,7,8,9,10,  
   11,12,13,14,15....20.  
   21,22,23,24,25....30  
   .....    
   .................200]   

Это в 1d-списке

если мы выберем прямоугольники, скажем, 2 * 3 (w * h) Первый будет содержать 1,2,11,12,21,22 второй будет содержать 3,4,13,14,23,24 и т.д. до конца

Спасибо

  • 1
    Можете ли вы привести пример своего вклада и каким бы он был?
  • 0
    Что вы подразумеваете под "стандартными функциями"? Похоже, вы ищете определенный стиль кода. Почему бы не довольствоваться только кодом, который выполняет работу?
Показать ещё 5 комментариев
Теги:
list
slice

5 ответов

2
Лучший ответ
width = 6
height = 4
xs = range(1,25)
w = 3
h = 2

def subrect(x,y):
    pos = y*h*width+x*w
    return [xs[(pos+row*width):(pos+row*width+w)] for row in range(h)]

print [subrect(x,y) for y in range(height / h) for x in range(width / w)]

разбивает матрицу следующим образом:

 1  2  3     4  5  6
 7  8  9    10 11 12

13 14 15    16 17 18
19 20 21    22 23 24

EDIT: Или для примера, который вы дали...

width = 10
height = 20
xs = range(1,201)
w = 2
h = 3
2

Обратите внимание, что ваш вопрос указывает, что входной список равен 1D, но не дает указания на количество элементов для каждой логической строки; вы, кажется, волшебным образом подразумеваете, что это должно быть 10 элементов в строке.

Итак, учитывая 1D-список, количество логических элементов на строку, ширину и высоту запрошенных фрагментов, вы можете сделать:

def gettiles(list1d, row_items, width, height):
    o_row= 0
    row_count, remainder= divmod(len(list1d), row_items)
    if remainder != 0:
        raise RuntimeError("item count not divisible by %d" % row_items)
    if row_count % height != 0:
        raise RuntimeError("row count not divisible by height %d" % height)
    if row_items % width != 0:
        raise RuntimeError("row width not divisible by %d" % width)
    for o_row in xrange(0, row_count, height):
        for o_col in xrange(0, row_items, width):
            result= []
            top_left_index= o_row*row_items + o_col
            for off_row in xrange(height):
                for off_col in xrange(width):
                    result.append(list1d[top_left_index + off_row*row_items + off_col])
            yield result

>>> import pprint
>>> pprint.pprint(list(gettiles(range(100), 10, 2, 5)))
[[0, 1, 10, 11, 20, 21, 30, 31, 40, 41],
 [2, 3, 12, 13, 22, 23, 32, 33, 42, 43],
 [4, 5, 14, 15, 24, 25, 34, 35, 44, 45],
 [6, 7, 16, 17, 26, 27, 36, 37, 46, 47],
 [8, 9, 18, 19, 28, 29, 38, 39, 48, 49],
 [50, 51, 60, 61, 70, 71, 80, 81, 90, 91],
 [52, 53, 62, 63, 72, 73, 82, 83, 92, 93],
 [54, 55, 64, 65, 74, 75, 84, 85, 94, 95],
 [56, 57, 66, 67, 76, 77, 86, 87, 96, 97],
 [58, 59, 68, 69, 78, 79, 88, 89, 98, 99]]
1

Или это, что довольно просто.

def genMatrix(rows, cols, mylist):
   for x in xrange(rows):
      yield mylist[x*cols:x*cols+cols]

Результаты

>>> L = [1,1,1,1,2,2,2,2]
>>> list(genMatrix(2, 4, L))
[[1, 1, 1, 1], [2, 2, 2, 2]]
>>> L = [1,1,1,1,2,2,2,2,3,3,3,3]
>>> list(genMatrix(3, 4, L))
[[1, 1, 1, 1], [2, 2, 2, 2], [3, 3, 3, 3]]
1

Вот предложение (возможно, довольно неэффективное, но, похоже, работает):

def rect_slice(seq, cols, width, height):
    rows = len(seq) // cols
    for i in xrange(0, rows - rows % height, height):
        for j in xrange(0, cols - cols % width, width):
            yield [seq[k * cols + l] for k in xrange(i, i + height) for l in xrange(j, j + width)]

print list(rect_slice(range(1, 201), 10, 2, 3))
0

Если я правильно понял, у вас есть [1,1,1,1,2,2,2,2] и хотите [[1,1,1,1], [2,2,2,2]]? Если да, то просто:

L = [1,1,1,1,2,2,2,2]
w = 4
matrix = [L[:w], L[w:]]

по крайней мере для 2d.

или вы можете написать это для более общего решения:

def genMatrix(rows, cols, mylist):
   matrix = []
   for x in xrange(rows):
      row = []
      for y in xrange(cols):
         row.append(mylist[x*cols])
      matrix.append(row)
   return matrix

print genMatrix(2, 4, L) # => [[1,1,1,1], [2,2,2,2]]
L = [1,1,1,1,2,2,2,2,3,3,3,3]
print getnMatrix(3, 4, L) # => [[1,1,1,1], [2,2,2,2], [3,3,3,3]]

Ещё вопросы

Сообщество Overcoder
Наверх
Меню