python - roots - Encuentre índices de un valor en la matriz 2d

scipy optimize minimize in python

Question

Tengo una matriz de la forma,

mymatrix=[[1,2,3],[4,5,6],[7,8,9]]

Quiero obtener el índice de, digamos por ejemplo, 9 que está en (2,2).

Lo que he tratado de hacer hasta ahora.

for i,j in enumerate(mymatrix):
   for k,l in enumerate(j):
     if l==9:
         print i,k

¿Hay una mejor manera de hacer lo mismo? Optimización, ¿alguien? Gracias por adelantado.

Answer 1

Creo que puede encontrar útil, esclarecedor e incluso sorprendente lo siguiente:

Editar: Se movió el valor objetivo al centro de la matriz para simular la ubicación promedio si los datos son aleatorios y nivelan el campo de juego para los algoritmos que se detienen tan pronto como se encuentran.

También ejecutó sincronizaciones en Python 2 y 3 para comparar.

from __future__ import print_function
import sys
import timeit

setup = """
mymatrix=[[1,2,3],[4,9,6],[7,8,5]]  # moved target value to middle
val = 9
"""

statements = {
"Anuk (OP)": """
# finds all occurrences
found = []
for i,j in enumerate(mymatrix):
   for k,l in enumerate(j):
     if l==val:
         found.append((i,k))
""",

"Ryan Haining": """
# only finds first occurrence in each row
found = [(index, row.index(val)) for index, row in enumerate(mymatrix) if val in row]
""",

"martineau": """
# finds all occurrences
width = len(mymatrix[0])
found = []
posn = 0
for row in mymatrix:
    for col in row:
        if col == val:
            found.append((posn // width, posn % width))
        posn += 1
""",

"martineau #2": """
# finds all occurrences
width = len(mymatrix[0])
found = [(posn // width, posn % width)
         for posn,elem in enumerate(col for row in mymatrix for col in row)
            if elem == val]
""",

"mtahmed": """
# stops after it finds first occurrence
matrix_dim = len(mymatrix[0])
item_index = 0
for row in mymatrix:
    for i in row:
        if i == val:
            break
        item_index += 1
    if i == val:
        break
found = [(int(item_index / matrix_dim), item_index % matrix_dim)]
""",
}

N = 1000000
R = 3

timings = [
    (idea,
     min(timeit.repeat(statements[idea], setup=setup, repeat=R, number=N)),
    ) for idea in statements]

longest = max(len(t[0]) for t in timings)  # length of longest name

print(''fastest to slowest timings (Python {}.{}.{})/n''.format(*sys.version_info[:3]),
      ''  ({:,d} executions, best of {:d})/n''.format(N, R))

ranked = sorted(timings, key=lambda t: t[1])  # sort by speed (fastest first)
for timing in ranked:
    print("{:>{width}} : {:.6f} secs, rel speed {rel:>8.6f}x".format(
          timing[0], timing[1], rel=timing[1]/ranked[0][1], width=longest))

Muestra de salida:

fastest to slowest timings (Python 2.7.5)
   (1,000,000 executions, best of 3)

     mtahmed : 2.850508 secs, rel speed 1.000000x
   martineau : 3.684153 secs, rel speed 1.292455x
Ryan Haining : 8.391357 secs, rel speed 2.943811x
   Anuk (OP) : 14.014551 secs, rel speed 4.916510x
martineau #2 : 15.880949 secs, rel speed 5.571270x

fastest to slowest timings (Python 3.3.2)
   (1,000,000 executions, best of 3)

     mtahmed : 5.019435 secs, rel speed 1.000000x
   martineau : 5.217747 secs, rel speed 1.039509x
Ryan Haining : 5.705710 secs, rel speed 1.136723x
   Anuk (OP) : 8.317911 secs, rel speed 1.657141x
martineau #2 : 11.590270 secs, rel speed 2.309078x

Answer 2

Puede hacer esto en lugar de usar enumerar. NO estoy seguro si esto es más rápido.

matrix = [[1,2,3],[4,5,6],[7,8,9],[10,11,12]]
needle = 9

matrix_dim = len(matrix[0])
item_index = 0
for row in matrix:
    for i in row:
        if i == needle:
            break
        item_index += 1
    if i == needle:
        break

print(int(item_index / matrix_dim), item_index % matrix_dim)

Esto tomará exactamente el tiempo i * dim(matrix) + (j+1) donde el resultado de lo anterior es ij que podría ser O(n^2) en el peor de los casos.

Answer 3

Si convierte mymatrix a una matriz numpy, puede usar numpy.where para devolver los índices:

>>> import numpy as np
>>> mymatrix=[[1,2,3],[4,5,6],[7,8,9]]
>>> a = np.array(mymatrix)
>>> a
array([[1, 2, 3],
       [4, 5, 6],
       [7, 8, 9]])
>>> b = np.where(a==9)
>>> b
(array([2]), array([2]))
>>> mymatrix=[[1,2,3],[9,5,6],[7,8,9]]
>>> a = np.array(mymatrix)
>>> a
array([[1, 2, 3],
       [9, 5, 6],
       [7, 8, 9]])
>>> b = np.where(a==9)
>>> b
(array([1, 2]), array([0, 2]))

Answer 4

Si desea todas las ubicaciones en las que aparece el valor, puede usar la siguiente lista de comprensión con val set a lo que está buscando.

[(index, row.index(val)) for index, row in enumerate(mymatrix) if val in row]

por ejemplo:

>>> mymatrix=[[1,2,9],[4,9,6],[7,8,9]]
>>> val = 9
>>> [(index, row.index(val)) for index, row in enumerate(mymatrix) if val in row]
[(0, 2), (1, 1), (2, 2)]

EDITAR

No es realmente cierto que esto obtenga todas las ocurrencias, solo obtendrá la primera aparición del valor en una fila dada.

Answer 5

si desea encontrar el índice de todas las ocurrencias de un val o personaje en una lista 2d, este código puede ayudarlo y es legible. Tnq.

for i, e in enumerate(board):
    for j, ee in enumerate(e):
        if ''d'' in ee:
            print(i, j)

puedes encontrar múltiples ocurrencias también.