nan_to_num.hpp

Go to the documentation of this file.

#pragma once
 
#include <utility>
 
#include "NumCpp/Core/DtypeInfo.hpp"
#include "NumCpp/Core/Internal/StaticAsserts.hpp"
#include "NumCpp/Core/Internal/StlAlgorithms.hpp"
#include "NumCpp/Functions/isinf.hpp"
#include "NumCpp/Functions/isnan.hpp"
#include "NumCpp/NdArray.hpp"
 
namespace nc
{
    //============================================================================
    // Method Description:
    template<typename dtype>
    NdArray<dtype> nan_to_num(NdArray<dtype> inArray,
                              dtype          nan    = static_cast<dtype>(0.),
                              dtype          posInf = DtypeInfo<dtype>::max(),
                              dtype          negInf = DtypeInfo<dtype>::min())
    {
        STATIC_ASSERT_FLOAT(dtype);
 
        stl_algorithms::for_each(inArray.begin(),
                                 inArray.end(),
                                 [nan, posInf, negInf](dtype& value)
                                 {
                                     if (isnan(value))
                                     {
                                         value = nan;
                                     }
                                     else if (isinf(value))
                                     {
                                         if (value > static_cast<dtype>(0.))
                                         {
                                             value = posInf;
                                         }
                                         else
                                         {
                                             value = negInf;
                                         }
                                     }
                                 });
 
        return inArray;
    }
} // namespace nc