【Python】实现二维List的顺时针旋转

示例:输入  [[1,2,3,7],[4,5,6,9],[7,9,2,4]]

得到输出:  [[7,4,1],[9,5,2],[2,6,3],[4,9,7]]

方法一:使用List内置方法  O(logN^3)

for i in range(len(arr)):
    for j in range(len(arr[i])):
	if i == 0:
	    r_arr.append([arr[i][j]])
	else:
	    r_arr[j].insert(0,arr[i][j])

        思路非常简单:首先遍历List内的所有元素。
        其中第一行的元素最后将出现在新数组每一列的最后一位,第二行的元素将出现在新数组每一列的倒数第二位……以此类推。因此我们只需要遍历,并将元素依次向前插入即可。但此法时间复杂度过高。insert()方法本身自带N个时间复杂度。因此实际运行会非常慢。

        如果可以确定这个二维List是矩阵,那么我们就不用这么麻烦。下面的代码来自王神仙前辈:

def rotate90Angle(arr):
    result = []
    h = len(arr) - 1
    w = len(arr[0])  - 1
    for j in range(len(arr[0])):
        item = []
        for i in range(len(arr)):
            item.append(arr[h-i][j])
        result.append(item)
    return result

方法二:使用Numpy库中的数组方法

arr = np.array([[1,2,3,7],[4,5,6,9],[7,9,2,4]])  #用np定义数组

def rotaryR(arr):
    r_arr = arr.reshape(arr.size)
    r_arr = r_arr[::-1]
    r_arr = r_arr.reshape(arr.shape)
    r_arr = np.transpose(r_arr)[::-1]
    return r_arr

顺带一提逆时针旋转的方法如下:

def rotaryL(arr):
    r_arr = np.transpose(arr)
    r_arr = r_arr[::-1]
    return r_arr

想法也很简单,分别用transpose()方法和reshape()方法对数组进行序列上和形状上的改变即可。

你可能感兴趣的:(Python)