無法重寫代碼以列出理解python中圖像旋轉的代碼

Question

因此，在stackoverflow上找不到我的問題之后，這是如何將for循環重寫為列表推導式（在其中插入值的地方），我現在必須詢問是否可以重寫此代碼：

rotate = np.zeros((w,h,c), np.uint8) # create an empty image filled with zeros turned 90°

for y in xrange(h):
    for x in xrange(w):
        rotate[x][y] = img[y][x]

進入列表理解？ 我以為這樣的事情會起作用，但是沒有：

rotate[x][y] = img[y][x] for y in range(h) for x in range(w) 

在那之后，我只是嘗試添加索引和括號的各種組合，並且總是遇到一些語法錯誤。 僅作記錄，我知道opencv和numpy中都有用於旋轉圖像的函數，我只是想將for循環重寫為列表理解。

Answer 1

rotate = np.array([[img[y][x] for y in xrange(h)] for x in xrange(w)])