typestar

El argumento axis en Python

axis=0 baja por las filas, axis=1 cruza por las columnas.

import numpy as np

malla = np.array([[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]])

print(malla.sum(), malla.sum(axis=0), malla.sum(axis=1))
print(malla.mean(axis=1, keepdims=True))
print(malla / malla.sum(axis=1, keepdims=True))
print(malla.argmax(axis=0))

Cómo funciona

  1. Un agregado sobre el eje 0 da un valor por columna.
  2. keepdims=True conserva el rango, y eso le gusta al broadcasting.
  3. Omitir axis reduce el arreglo completo.

Palabras clave y builtins usados aquí

El intento, en números

Líneas
8
Caracteres a escribir
247
Tokens
97
Ritmo de tres estrellas
105 tpm

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

Escribe este fragmento

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

← Anterior Siguiente →