1030. 距离顺序排列矩阵单元格

为了便于理解，我们在这里先假设一个3*4的数组，并假设r0=1,c0=1
此部分代码如下：

import numpy as np
arr = np.arange(0,12)   #生成0到11的一维数组
arr = arr.reshape((3,4)) # 将一维数组重新排列为3*4的数组
r0 = 1
c0 = 1 

我们可以打印出arr和arr[r0][c0]先看看


此后，我们需要对计算每一个元素对于（r0,c0)的距离，因为这是后面排序的依据，所以我们可以建立一个表格类型的变量（命名为lis）对数据进行储存。

lis = []

然后遍历每个元素，并计算与（r0,c0）的曼哈顿距离。

for i in range(arr.shape[0]):
    for j in range(arr.shape[1]):
        d = abs(i-r0) + abs(j-c0)
        lis.append([i,j])

打印lis看看：

然后就是依据lis进行排序。
此处借鉴题解中的代码，使用list的sort()方法并将匿名函数作为key实现排序。

lis.sort(key = lambda x: abs(x[0]-r0) + abs(x[1]-c0))

解释这里为什么会有x[0]、x[1]，因为此处的x其实就相当于lis[0]，lis[1]…

以lis[1]为例，此处的x[0]就代表的是第一个元素0，x[1]代表的是第二个元素1.

举例子的整体代码：

import numpy as np
arr = np.arange(0,12)
arr = arr.reshape((3,4))
r0 = 1
c0 = 1 
lis = []
for i in range(arr.shape[0]):
    for j in range(arr.shape[1]):
        d = abs(i-r0) + abs(j-c0)
        lis.append([i,j])
lis.sort(key = lambda x: abs(x[0]-r0) + abs(x[1]-c0))        

改成题目要求格式（抽象化）：

class Solution(object):
    def allCellsDistOrder(self, R, C, r0, c0):
    	res = []
		for i in range(arr.shape[0]):
   	   	    for j in range(arr.shape[1]):
                d = abs(i-r0) + abs(j-c0)
                lis.append([i,j])
    	return lis.sort(key = lambda x: abs(x[0]-r0) + abs(x[1]-c0))

其实上面代码可以一步到位（看的一位大佬的）：

class Solution(object):
    def allCellsDistOrder(self, R, C, r0, c0):
        return sorted([[i, j] for i in range(R) for j in range(C)], key=lambda x: abs(x[0] - r0) + abs(x[1] - c0))