Indexación avanzada en Python

Indexar con arreglos de posiciones, que copia en lugar de dar vistas.

import numpy as np

malla = np.arange(20).reshape(4, 5)

print(malla[[0, 3, 0]])
print(malla[[0, 1, 2], [4, 3, 2]])

puntajes = np.array([88, 104, 91, 79])
orden = np.argsort(puntajes)[::-1]
print(orden, puntajes[orden])
print(np.argmax(puntajes), np.argmin(puntajes))

Cómo funciona

1. Un arreglo de enteros elige filas en cualquier orden, con repeticiones permitidas.
2. Dos arreglos de índices eligen elementos individuales por pares.
3. argsort da la permutación que ordenaría el arreglo.

Palabras clave y builtins usados aquí

• as
• print

El intento, en números

Líneas

11

Caracteres a escribir

268

Tokens

108

Ritmo de tres estrellas

105 tpm

Al ritmo de tres estrellas de 105 tokens por minuto, este intento toma unos 62 segundos.

Escribe este fragmento

Paso 2 de 3 en Indexación; paso 14 de 22 en NumPy.

← Anterior Siguiente →