find.hpp

Go to the documentation of this file.

 
#pragma once
 
#include <limits>
 
#include "NumCpp/Core/Types.hpp"
#include "NumCpp/NdArray.hpp"
 
namespace nc
{
    //============================================================================
    // Method Description:
    inline NdArray<uint32> find(const NdArray<bool>& mask, uint32 n = std::numeric_limits<uint32>::max())
    {
        NdArray<uint32> indices = mask.flatnonzero();
 
        if (indices.size() <= n)
        {
            return indices;
        }
 
        return indices[Slice(0, n)];
    }
    inline NdArray<uint32> find(const NdArray<bool>& mask, uint32 n = std::numeric_limits<uint32>::max()) {…}
} // namespace nc