2010-07-12 193 views
1

如果我有一个表示'n'元素(每个元素是一个单字节)的列表,它表示一个矩形的2d矩阵,我怎样才能将它分成说w * h的矩形,从第一个元素开始列表中的,只是使用Python标准功能切片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] 

这些都是在一维列表

如果我们选择矩形的说,2 * 3(W * H) 第一将包含1,2,11,12,21,22 该se COND将包含3,4,13,14,23,24等,直到最后

感谢

+1

你可以给你的输入样本和你会想要输出? – bwawok 2010-07-12 15:22:06

+0

你是什么意思“标准功能”?这听起来像你正在寻找一种特定的代码风格。为什么不解决完成工作的代码? – 2010-07-12 15:26:18

+0

是输入'list' 1D?如果是这样,在行 - 主要或列主要秩序? – MAK 2010-07-12 15:32:37

回答

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 

编辑:还是你给的例子...

width = 10 
height = 20 
xs = range(1,201) 
w = 2 
h = 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]] 
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)) 
2

请注意,您的问题指定输入列表是一维的,但没有给出指示到多少个项目每个逻辑行;你似乎神奇地暗示它应该是每行10个项目。

所以,对于一个一维列表,每行逻辑的项目,宽度和要求的瓷砖高度的数量,你可以这样做:

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]]