Ennumpy Las dimensiones de la matriz resultante varían en el tiempo de ejecución. A menudo hay confusión entre una matriz 1d y una matriz 2d con 1 columna. En un caso puedo iterar sobre las columnas, en el otro caso no puedo.

¿Cómo resuelves elegantemente ese problema? Para evitar ensuciar mi código conif comprobando la dimensionalidad de las declaraciones, uso esta función:

def reshape_to_vect(ar):
    if len(ar.shape) == 1:
      return ar.reshape(ar.shape[0],1)
    return ar

Sin embargo, esto se siente poco elegante y costoso. ¿Hay una mejor solución?