NumCpp  2.7.0
A Templatized Header Only C++ Implementation of the Python NumPy Library
nansum.hpp
Go to the documentation of this file.
1 #pragma once
29 
32 #include "NumCpp/Core/Types.hpp"
33 #include "NumCpp/Functions/sum.hpp"
34 #include "NumCpp/NdArray.hpp"
35 
36 #include <cmath>
37 
38 namespace nc
39 {
40  //============================================================================
41  // Method Description:
51  template<typename dtype>
52  NdArray<dtype> nansum(const NdArray<dtype>& inArray, Axis inAxis = Axis::NONE)
53  {
54  STATIC_ASSERT_FLOAT(dtype);
55 
56  NdArray<dtype> arrayCopy(inArray);
57  stl_algorithms::for_each(arrayCopy.begin(), arrayCopy.end(),
58  [](dtype& value) noexcept -> void
59  {
60  if (std::isnan(value)) { value = static_cast<dtype>(0); };
61  });
62 
63  return sum(arrayCopy, inAxis);
64  }
65 } // namespace nc
#define STATIC_ASSERT_FLOAT(dtype)
Definition: StaticAsserts.hpp:43
Holds 1D and 2D arrays, the main work horse of the NumCpp library.
Definition: NdArrayCore.hpp:72
iterator end() noexcept
Definition: NdArrayCore.hpp:1474
iterator begin() noexcept
Definition: NdArrayCore.hpp:1166
void for_each(InputIt first, InputIt last, UnaryFunction f)
Definition: StlAlgorithms.hpp:213
Definition: Coordinate.hpp:45
Axis
Enum To describe an axis.
Definition: Types.hpp:46
NdArray< dtype > nansum(const NdArray< dtype > &inArray, Axis inAxis=Axis::NONE)
Definition: nansum.hpp:52
NdArray< dtype > sum(const NdArray< dtype > &inArray, Axis inAxis=Axis::NONE)
Definition: sum.hpp:46