NumCpp  2.6.2
A Templatized Header Only C++ Implementation of the Python NumPy Library
nan_to_num.hpp
Go to the documentation of this file.
1 
28 #pragma once
29 
30 #include "NumCpp/NdArray.hpp"
36 
37 #include <utility>
38 
39 namespace nc
40 {
41  //============================================================================
42  // Method Description:
55  template<typename dtype>
57  dtype nan = static_cast<dtype>(0.0),
58  dtype posInf = DtypeInfo<dtype>::max(),
59  dtype negInf = DtypeInfo<dtype>::min())
60  {
61  STATIC_ASSERT_FLOAT(dtype);
62 
63  stl_algorithms::for_each(inArray.begin(), inArray.end(),
64  [nan, posInf, negInf](dtype& value)
65  {
66  if (isnan(value))
67  {
68  value = nan;
69  }
70  else if (isinf(value))
71  {
72  if (value > static_cast<dtype>(0.0))
73  {
74  value = posInf;
75  }
76  else
77  {
78  value = negInf;
79  }
80  }
81  }
82  );
83 
84  return inArray;
85  }
86 } // namespace nc
#define STATIC_ASSERT_FLOAT(dtype)
Definition: StaticAsserts.hpp:43
Holds info about the dtype.
Definition: DtypeInfo.hpp:41
Holds 1D and 2D arrays, the main work horse of the NumCpp library.
Definition: NdArrayCore.hpp:72
iterator end() noexcept
Definition: NdArrayCore.hpp:1558
iterator begin() noexcept
Definition: NdArrayCore.hpp:1214
const double nan
NaN.
Definition: Constants.hpp:44
void for_each(InputIt first, InputIt last, UnaryFunction f)
Definition: StlAlgorithms.hpp:213
Definition: Coordinate.hpp:45
NdArray< dtype > nan_to_num(NdArray< dtype > inArray, dtype nan=static_cast< dtype >(0.0), dtype posInf=DtypeInfo< dtype >::max(), dtype negInf=DtypeInfo< dtype >::min())
Definition: nan_to_num.hpp:56
bool isinf(dtype inValue) noexcept
Definition: isinf.hpp:51