NumCpp  2.8.0
A Templatized Header Only C++ Implementation of the Python NumPy Library
count_nonzero.hpp
Go to the documentation of this file.
1 #pragma once
29 
32 #include "NumCpp/Core/Shape.hpp"
33 #include "NumCpp/Core/Types.hpp"
34 #include "NumCpp/NdArray.hpp"
35 
36 namespace nc
37 {
38  //============================================================================
39  // Method Description:
48  template<typename dtype>
50  {
52 
53  switch (inAxis)
54  {
55  case Axis::NONE:
56  {
57  NdArray<uint32> count = { inArray.size() -
58  static_cast<uint32>(
59  stl_algorithms::count(inArray.cbegin(), inArray.cend(), dtype{ 0 })) };
60  return count;
61  }
62  case Axis::COL:
63  {
64  Shape inShape = inArray.shape();
65 
66  NdArray<uint32> returnArray(1, inShape.rows);
67  for (uint32 row = 0; row < inShape.rows; ++row)
68  {
69  returnArray(0, row) =
70  inShape.cols -
71  static_cast<uint32>(stl_algorithms::count(inArray.cbegin(row), inArray.cend(row), dtype{ 0 }));
72  }
73 
74  return returnArray;
75  }
76  case Axis::ROW:
77  {
78  NdArray<dtype> inArrayTranspose = inArray.transpose();
79  Shape inShapeTransposed = inArrayTranspose.shape();
80  NdArray<uint32> returnArray(1, inShapeTransposed.rows);
81  for (uint32 row = 0; row < inShapeTransposed.rows; ++row)
82  {
83  returnArray(0, row) =
84  inShapeTransposed.cols - static_cast<uint32>(stl_algorithms::count(inArrayTranspose.cbegin(row),
85  inArrayTranspose.cend(row),
86  dtype{ 0 }));
87  }
88 
89  return returnArray;
90  }
91  default:
92  {
93  THROW_INVALID_ARGUMENT_ERROR("Unimplemented axis type.");
94  return {}; // get rid of compiler warning
95  }
96  }
97  }
98 } // namespace nc
#define THROW_INVALID_ARGUMENT_ERROR(msg)
Definition: Error.hpp:36
#define STATIC_ASSERT_ARITHMETIC_OR_COMPLEX(dtype)
Definition: StaticAsserts.hpp:49
Holds 1D and 2D arrays, the main work horse of the NumCpp library.
Definition: NdArrayCore.hpp:72
size_type size() const noexcept
Definition: NdArrayCore.hpp:4289
const_iterator cbegin() const noexcept
Definition: NdArrayCore.hpp:1221
Shape shape() const noexcept
Definition: NdArrayCore.hpp:4276
NdArray< dtype > transpose() const
Definition: NdArrayCore.hpp:4650
const_iterator cend() const noexcept
Definition: NdArrayCore.hpp:1529
A Shape Class for NdArrays.
Definition: Core/Shape.hpp:41
uint32 rows
Definition: Core/Shape.hpp:44
uint32 cols
Definition: Core/Shape.hpp:45
std::iterator_traits< InputIt >::difference_type count(InputIt first, InputIt last, const T &value) noexcept
Definition: StlAlgorithms.hpp:121
Definition: Coordinate.hpp:45
Axis
Enum To describe an axis.
Definition: Types.hpp:47
NdArray< uint32 > count_nonzero(const NdArray< dtype > &inArray, Axis inAxis=Axis::NONE)
Definition: count_nonzero.hpp:49
std::uint32_t uint32
Definition: Types.hpp:40