Estoy tratando de ordenar un ndarray numpy en el sentido de las agujas del reloj (en orden ascendente). Puede entenderlo como tomar todos los valores de la matriz, ordenarlos y luego colocarlos en una espiral en el sentido de las agujas del reloj en una nueva matriz con la misma forma. Las direcciones de la espiral en el sentido de las agujas del reloj se muestran a continuación:

Por ejemplo, supongamos que tengo una matriz de

 import numpy as np a1 = np.array(([2, 4, 6], [1, 5, 3], [7, 9, 8]))

La salida esperada es

 np.array([[1, 2, 3], [8, 9, 4], [7, 6, 5]])

Si tengo una matriz de

 a2 = np.array(([2, 4, 6], [1, 5, 3], [7, 9, 8], [12, 11, 10]))

La salida esperada es

 np.array([[1, 2, 3], [10, 11, 4], [9, 12, 5], [8, 7, 6]])

Lo que he probado hasta ahora

Mi idea es rastrear el índice de fila x y el índice de columna y del iterador en movimiento y el índice actual cur de la lista aplanada ordenada sa . El número de filas por recorrer ( lenr ) y el número de columnas por recorrer ( lenc ) se restan en 1 una vez que el iterador en movimiento pasa lenr filas horizontalmente y lenc columnas verticalmente. Aquí está la función que logré escribir:

 def clockwise_sorted(a, key=None, reverse=False): nr, nc = a.shape res = a.tolist() sa = a.ravel().tolist() if key is None: sa.sort(reverse=reverse) else: sa.sort(key=key, reverse=reverse) res[0] = sa[:nc] cur, lenr, lenc = nc, nr - 1, nc - 1 x, y = 0, nc - 1 while (lenc > 0 and lenr > 0): # go down, then go left for _ in range(lenr): x += 1 res[x][y] = sa[cur] cur += 1 for _ in range(lenc): y -= 1 res[x][y] = sa[cur] cur += 1 lenr -= 1 lenc -= 1 # go up, then go right for _ in range(lenr): x -= 1 res[x][y] = sa[cur] cur += 1 for _ in range(lenc): y += 1 res[x][y] = sa[cur] cur += 1 lenr -= 1 lenc -= 1 return np.array(res)

Para matrices cuadradas, mi código funciona bien:

 print(clockwise_sorted(a1)) #[[1 2 3] # [8 9 4] # [7 6 5]] #%timeit 5.98 µs ± 413 ns per loop (mean ± std. dev. of 7 runs, 100000 loops each)

Pero no funciona para matrices no cuadradas:

 print(clockwise_sorted(a2)) #[[ 1 2 3] # [10 11 4] # [ 9 9 5] # [ 8 7 6]]

Claramente, pierde el 12 en [2,1] .

Tenga en cuenta que no espero que entienda y corrija mi código. No me gusta mucho mi código porque siento que debería haber más formas numpyic de hacer esto. Por lo tanto, quiero ver algunas soluciones numpyic. Sin embargo, sería bueno si alguien puede comentar sobre lo que está mal con mi código.