NumCpp  2.9.0
A Templatized Header Only C++ Implementation of the Python NumPy Library
concatenate.hpp
Go to the documentation of this file.
1 #pragma once
29 
30 #include <initializer_list>
31 
33 #include "NumCpp/Core/Shape.hpp"
34 #include "NumCpp/Core/Types.hpp"
37 
38 namespace nc
39 {
40  //============================================================================
41  // Method Description:
50  template<typename dtype>
51  NdArray<dtype> concatenate(const std::initializer_list<NdArray<dtype>>& inArrayList, Axis inAxis = Axis::NONE)
52  {
53  switch (inAxis)
54  {
55  case Axis::NONE:
56  {
57  uint32 finalSize = 0;
58  for (auto& ndarray : inArrayList)
59  {
60  finalSize += ndarray.size();
61  }
62 
63  NdArray<dtype> returnArray(1, finalSize);
64  uint32 offset = 0;
65  for (auto& ndarray : inArrayList)
66  {
67  stl_algorithms::copy(ndarray.cbegin(), ndarray.cend(), returnArray.begin() + offset);
68  offset += ndarray.size();
69  }
70 
71  return returnArray;
72  }
73  case Axis::ROW:
74  {
75  return row_stack(inArrayList);
76  }
77  case Axis::COL:
78  {
79  return column_stack(inArrayList);
80  }
81  default:
82  {
83  THROW_INVALID_ARGUMENT_ERROR("Unimplemented axis type.");
84  return {}; // get rid of compiler warning
85  }
86  }
87  }
88 } // namespace nc
#define THROW_INVALID_ARGUMENT_ERROR(msg)
Definition: Error.hpp:36
Holds 1D and 2D arrays, the main work horse of the NumCpp library.
Definition: NdArrayCore.hpp:72
iterator begin() noexcept
Definition: NdArrayCore.hpp:1171
OutputIt copy(InputIt first, InputIt last, OutputIt destination) noexcept
Definition: StlAlgorithms.hpp:99
Definition: Coordinate.hpp:45
NdArray< dtype > column_stack(const std::initializer_list< NdArray< dtype >> &inArrayList)
Definition: column_stack.hpp:49
NdArray< dtype > row_stack(const std::initializer_list< NdArray< dtype >> &inArrayList)
Definition: row_stack.hpp:49
Axis
Enum To describe an axis.
Definition: Types.hpp:47
NdArray< dtype > concatenate(const std::initializer_list< NdArray< dtype >> &inArrayList, Axis inAxis=Axis::NONE)
Definition: concatenate.hpp:51
std::uint32_t uint32
Definition: Types.hpp:40