es.davy.ai

Preguntas y respuestas de programación confiables

¿Tienes una pregunta?

Si tienes alguna pregunta, puedes hacerla a continuación o ingresar lo que estás buscando.

Tag: AUTO-VECTORIZATION

Numpy: Condiciones anidadas en dos columnas de Pandas

Introducción: Tengo un dataframe de pandas con las columnas df = [openDate, high, low, open, close, volume, is_eligible]. En este dataframe: openDate es de tipo fecha y hora, high, low, open, close, volume son de tipo int64, is_eligible es de tipo booleano. Declaración del problema: Quiero añadir una columna más, . . . Read more

Función verdaderamente vectorizar para numpy array en Python.

Tengo la siguiente función, que toma dos arreglos 1D de numpy q_i y q_j, realiza algunos cálculos (incluyendo calcular la norma de su diferencia) y devuelve un arreglo de numpy: import numpy as np import numpy.linalg as lin def coulomb(q_i, q_j, c1=0.8, c2=0.2, k=0.5): “”” Parametros ———- q_i : numpy.ndarray . . . Read more

Implementar scatter max con numpy o pytorch para un array bidimensional.

Quiero implementar una versión vectorizada de la siguiente función utilizando numpy o pytorch: def scatter_max_2(src, index, out): src_shape = src.shape for i in range(src_shape[0]): for j in range(src_shape[1]): out[i][index[i][j]] = max(out[i][index[i][j]], src[i][j]) return out src = torch.tensor([[2, 0, 1, 4, 3], [0, 2, 1, 3, 4]]) index = torch.tensor([[4, 5, . . . Read more