2010-07-12 203 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]]