NumCpp  2.12.1
A Templatized Header Only C++ Implementation of the Python NumPy Library
nanargmax.hpp
Go to the documentation of this file.
1
28#pragma once
29
30#include <cmath>
31
35#include "NumCpp/Core/Types.hpp"
37#include "NumCpp/NdArray.hpp"
38
39namespace nc
40{
41 //============================================================================
42 // Method Description:
51 template<typename dtype>
53 {
55
56 NdArray<dtype> arrayCopy(inArray);
58 arrayCopy.end(),
59 [](dtype& value) noexcept -> void
60 {
61 if (std::isnan(value))
62 {
63 value = DtypeInfo<dtype>::min();
64 };
65 });
66
67 return argmax(arrayCopy, inAxis);
68 }
69} // namespace nc
#define STATIC_ASSERT_FLOAT(dtype)
Definition: StaticAsserts.hpp:50
Holds 1D and 2D arrays, the main work horse of the NumCpp library.
Definition: NdArrayCore.hpp:139
iterator end() noexcept
Definition: NdArrayCore.hpp:1623
iterator begin() noexcept
Definition: NdArrayCore.hpp:1315
void for_each(InputIt first, InputIt last, UnaryFunction f)
Definition: StlAlgorithms.hpp:225
Definition: Cartesian.hpp:40
Axis
Enum To describe an axis.
Definition: Enums.hpp:36
NdArray< uint32 > nanargmax(const NdArray< dtype > &inArray, Axis inAxis=Axis::NONE)
Definition: nanargmax.hpp:52
NdArray< uint32 > argmax(const NdArray< dtype > &inArray, Axis inAxis=Axis::NONE)
Definition: argmax.hpp:46