NumCpp  1.0
A C++ implementation of the Python Numpy library
find.hpp
Go to the documentation of this file.
1 #pragma once
30 
31 #include "NumCpp/Core/Types.hpp"
32 #include "NumCpp/NdArray.hpp"
33 
34 #include <limits>
35 
36 namespace nc
37 {
38  //============================================================================
39  // Method Description:
49  {
50  NdArray<uint32> indices = mask.flatnonzero();
51 
52  if (indices.size() <= n)
53  {
54  return indices;
55  }
56  else
57  {
58  return indices[Slice(0, n)];
59  }
60  }
61 }
nc::find
NdArray< uint32 > find(const NdArray< bool > &mask, uint32 n=std::numeric_limits< uint32 >::max())
Definition: find.hpp:48
nc::NdArray
Holds 1D and 2D arrays, the main work horse of the NumCpp library.
Definition: NdArrayCore.hpp:75
nc::uint32
std::uint32_t uint32
Definition: Types.hpp:41
NdArray.hpp
nc::NdArray::size
size_type size() const noexcept
Definition: NdArrayCore.hpp:4310
nc
Definition: Coordinate.hpp:45
nc::max
NdArray< dtype > max(const NdArray< dtype > &inArray, Axis inAxis=Axis::NONE)
Definition: max.hpp:46
nc::NdArray::flatnonzero
NdArray< uint32 > flatnonzero() const
Definition: NdArrayCore.hpp:2741
Types.hpp
nc::Slice
A Class for slicing into NdArrays.
Definition: Slice.hpp:44