MultiIndex ========== .. cpp:class:: pandas::MultiIndex Index class for axis labels in pandas data structures. Example ------- .. code-block:: cpp #include using namespace pandas; // Create MultiIndex MultiIndex idx({1, 2, 3}, "my_index"); size_t len = idx.size(); Constructors ------------ .. list-table:: :widths: 55 25 20 :header-rows: 1 * - Signature - Location - Example * - ``MultiIndex(std::vector>&& levels, std::vector>&& codes, std::vector> names = {}, bool verify_integrity = true)`` - pd_multiindex.h:88 - :ref:`View ` * - ``MultiIndex(const MultiIndex& other)`` - pd_multiindex.h:125 - :ref:`View ` * - ``MultiIndex(MultiIndex&& other) noexcept = default`` - pd_multiindex.h:149 - :ref:`View ` Construction ------------ .. list-table:: :widths: 40 20 15 25 :header-rows: 1 * - Signature - Return Type - Location - Example * - ``static MultiIndex from_arrays(const std::vector>& arrays, const std::vector>& names = {}, const std::optional& sortorder = std::nullopt)`` - MultiIndex - pd_multiindex.h:186 - :ref:`View ` * - ``static MultiIndex from_arrays(std::initializer_list> arrays, const std::vector>& names = {}, const std::optional& sortorder = std::nullopt)`` - MultiIndex - pd_multiindex.h:240 - :ref:`View ` * - ``static MultiIndex from_arrays(const std::vector& arr1, const std::vector& arr2, std::initializer_list names)`` - MultiIndex - pd_multiindex.h:256 - :ref:`View ` * - ``static MultiIndex from_frame(const std::vector>& df, const std::vector>& names = {}, const std::optional& sortorder = std::nullopt)`` - MultiIndex - pd_multiindex.h:1805 - :ref:`View ` * - ``static MultiIndex from_mixed_arrays( const std::vector>& int_arrays, const std::vector>& str_arrays, const std::string& level_types, const std::vector>& names = {})`` - static MultiIndex - pd_multiindex.h:278 - :ref:`View ` * - ``static MultiIndex from_product(const std::vector>& iterables, const std::vector>& names = {}, const std::optional& sortorder = std::nullopt)`` - MultiIndex - pd_multiindex.h:420 - :ref:`View ` * - ``static MultiIndex from_product(std::initializer_list> iterables, const std::vector>& names = {}, const std::optional& sortorder = std::nullopt)`` - MultiIndex - pd_multiindex.h:470 - :ref:`View ` * - ``static MultiIndex from_tuples(const std::vector>& tuples, const std::vector>& names = {}, const std::optional& sortorder = std::nullopt)`` - MultiIndex - pd_multiindex.h:373 - :ref:`View ` * - ``static MultiIndex from_tuples(std::initializer_list> tuples, const std::vector>& names = {}, const std::optional& sortorder = std::nullopt)`` - MultiIndex - pd_multiindex.h:406 - :ref:`View ` Indexing / Selection -------------------- .. list-table:: :widths: 40 20 15 25 :header-rows: 1 * - Signature - Return Type - Location - Example * - ``numpy::NDArray get_indexer(const MultiIndex& target, const std::string& method = "", std::optional limit = std::nullopt, std::optional tolerance = std::nullopt) const`` - numpy::NDArray - pd_multiindex.h:922 - :ref:`View ` * - ``numpy::NDArray get_indexer_for(const MultiIndex& target) const`` - numpy::NDArray - pd_multiindex.h:1817 - :ref:`View ` * - ``get_indexer_non_unique(const MultiIndex& target) const`` - - pd_multiindex.h:1827 - :ref:`View ` * - ``const IndexBase& get_level(size_t level) const`` - const IndexBase& - pd_multiindex.h:688 - :ref:`View ` * - ``const std::vector& get_level_categories(size_t level) const`` - const std::vector& - pd_multiindex.h:559 - * - ``std::unique_ptr get_level_values(int level) const`` - std::unique_ptr - pd_multiindex.h:614 - :ref:`View ` * - ``std::unique_ptr get_level_values(const std::string& level_name) const`` - std::unique_ptr - pd_multiindex.h:643 - :ref:`View ` * - ``std::vector get_level_values_str(size_t level) const`` - std::vector - pd_multiindex.h:666 - :ref:`View ` * - ``Index get_level_values_typed(size_t level) const`` - Index - pd_multiindex.h:1857 - * - ``std::variant> get_loc(const std::vector& key) const`` - std::variant> - pd_multiindex.h:882 - :ref:`View ` * - ``std::pair, MultiIndex> get_loc_level( const std::string& key, size_t level = 0, bool drop_level = true) const`` - std::pair, MultiIndex> - pd_multiindex.h:1886 - :ref:`View ` * - ``std::optional get_loc_string(const std::string& key) const`` - std::optional - pd_multiindex.h:3263 - :ref:`View ` * - ``numpy::NDArray get_locs(const std::vector>& seq) const`` - numpy::NDArray - pd_multiindex.h:1916 - :ref:`View ` * - ``std::optional get_name(size_t level) const`` - std::optional - pd_multiindex.h:698 - :ref:`View ` * - ``size_t get_slice_bound(const std::vector& label, const std::string& side) const`` - size_t - pd_multiindex.h:1944 - :ref:`View ` * - ``std::string get_string(size_t i) const`` - std::string - pd_multiindex.h:1083 - :ref:`View ` * - ``std::vector get_tuple_str(size_t index) const`` - std::vector - pd_multiindex.h:955 - :ref:`View ` * - ``std::pair> take( const std::vector& indices, int axis = 0, bool allow_fill = false, std::nullptr_t fill_value = nullptr) const`` - std::pair> - pd_multiindex.h:1193 - :ref:`View ` * - ``MultiIndex where(const numpy::NDArray& cond, const std::vector& other) const`` - MultiIndex - pd_multiindex.h:3075 - :ref:`View ` Data Manipulation ----------------- .. list-table:: :widths: 40 20 15 25 :header-rows: 1 * - Signature - Return Type - Location - Example * - ``MultiIndex drop(const std::vector>& labels, std::optional> codes = std::nullopt, std::optional level = std::nullopt, const std::string& errors = "raise") const`` - MultiIndex - pd_multiindex.h:1509 - :ref:`View ` * - ``MultiIndex drop_duplicates(const std::string& keep = "first") const`` - MultiIndex - pd_multiindex.h:1545 - :ref:`View ` * - ``MultiIndex droplevel(size_t level) const`` - MultiIndex - pd_multiindex.h:743 - :ref:`View ` * - ``MultiIndex dropna(const std::string& how = "any") const`` - MultiIndex - pd_multiindex.h:1597 - :ref:`View ` * - ``MultiIndex insert(size_t loc, const std::vector& item) const`` - MultiIndex - pd_multiindex.h:2020 - :ref:`View ` * - ``reindex(const MultiIndex& target, const std::string& method = "", std::optional level = std::nullopt, std::optional limit = std::nullopt, std::optional tolerance = std::nullopt) const`` - - pd_multiindex.h:2521 - :ref:`View ` * - ``MultiIndex rename(const std::vector>& new_names, std::optional level = std::nullopt, bool inplace = false, const std::optional>& names = std::nullopt) const`` - MultiIndex - pd_multiindex.h:2543 - :ref:`View ` * - ``MultiIndex reorder_levels(const std::vector& order) const`` - MultiIndex - pd_multiindex.h:801 - :ref:`View ` * - ``MultiIndex set_names(const std::vector>& names, std::optional level = std::nullopt, bool inplace = false) const`` - MultiIndex - pd_multiindex.h:716 - :ref:`View ` * - ``MultiIndex swaplevel(size_t i, size_t j) const`` - MultiIndex - pd_multiindex.h:771 - :ref:`View ` Missing Data ------------ .. list-table:: :widths: 40 20 15 25 :header-rows: 1 * - Signature - Return Type - Location - Example * - ``MultiIndex fillna(const std::vector& value, const std::string& downcast = "") const`` - MultiIndex - pd_multiindex.h:1734 - :ref:`View ` * - ``numpy::NDArray isna() const`` - numpy::NDArray - pd_multiindex.h:2218 - :ref:`View ` * - ``numpy::NDArray isnull() const`` - numpy::NDArray - pd_multiindex.h:2229 - :ref:`View ` * - ``numpy::NDArray notna() const`` - numpy::NDArray - pd_multiindex.h:2445 - :ref:`View ` * - ``numpy::NDArray notnull() const`` - numpy::NDArray - pd_multiindex.h:2456 - :ref:`View ` Statistics ---------- .. list-table:: :widths: 40 20 15 25 :header-rows: 1 * - Signature - Return Type - Location - Example * - ``std::vector max() const`` - std::vector - pd_multiindex.h:2390 - :ref:`View ` * - ``std::vector min() const`` - std::vector - pd_multiindex.h:2434 - :ref:`View ` * - ``size_t nunique() const`` - size_t - pd_multiindex.h:2463 - :ref:`View ` Aggregation ----------- .. list-table:: :widths: 40 20 15 25 :header-rows: 1 * - Signature - Return Type - Location - Example * - ``groupby(const std::vector& values) const`` - - pd_multiindex.h:1970 - :ref:`View ` * - ``MultiIndex map(Func mapper, const std::string& na_action = "") const`` - MultiIndex - pd_multiindex.h:2372 - :ref:`View ` Comparison ---------- .. list-table:: :widths: 40 20 15 25 :header-rows: 1 * - Signature - Return Type - Location - Example * - ``int compare_tuples(size_t idx_a, size_t idx_b) const`` - int - pd_multiindex.h:3151 - * - ``bool equal_levels(const MultiIndex& other) const`` - bool - pd_multiindex.h:1650 - :ref:`View ` * - ``bool equals(const MultiIndex& other) const`` - bool - pd_multiindex.h:1674 - :ref:`View ` * - ``const std::vector>& levels() const`` - const std::vector>& - pd_multiindex.h:520 - :ref:`View ` * - ``std::vector levshape() const`` - std::vector - pd_multiindex.h:491 - :ref:`View ` Sorting ------- .. list-table:: :widths: 40 20 15 25 :header-rows: 1 * - Signature - Return Type - Location - Example * - ``numpy::NDArray argsort() const`` - numpy::NDArray - pd_multiindex.h:1153 - :ref:`View ` * - ``size_t searchsorted(const std::vector& value, const std::string& side = "left", const std::optional>& sorter = std::nullopt) const`` - size_t - pd_multiindex.h:2605 - :ref:`View ` * - ``MultiIndex sort_values(bool ascending = true, const std::string& na_position = "last", bool return_indexer = false, std::nullptr_t key = nullptr) const`` - MultiIndex - pd_multiindex.h:2758 - :ref:`View ` Reshaping --------- .. list-table:: :widths: 40 20 15 25 :header-rows: 1 * - Signature - Return Type - Location - Example * - ``std::map> to_frame(bool index = true, bool allow_duplicates = false, const std::optional>& name = std::nullopt) const`` - std::map> - pd_multiindex.h:2838 - :ref:`View ` * - ``MultiIndex transpose(bool copy_data = false) const`` - MultiIndex - pd_multiindex.h:2902 - :ref:`View ` Combining --------- .. list-table:: :widths: 40 20 15 25 :header-rows: 1 * - Signature - Return Type - Location - Example * - ``MultiIndex append(const MultiIndex& other) const`` - MultiIndex - pd_multiindex.h:1234 - :ref:`View ` * - ``join(const MultiIndex& other, const std::string& how = "left", std::optional level = std::nullopt, bool return_indexers = true, bool sort = false) const`` - - pd_multiindex.h:2254 - :ref:`View ` Time Series ----------- .. list-table:: :widths: 40 20 15 25 :header-rows: 1 * - Signature - Return Type - Location - Example * - ``std::optional> asof(const std::vector& key, const std::optional>& label = std::nullopt) const`` - std::optional> - pd_multiindex.h:1329 - :ref:`View ` * - ``numpy::NDArray asof_locs( const std::vector>& where, const numpy::NDArray& mask) const`` - numpy::NDArray - pd_multiindex.h:1359 - :ref:`View ` * - ``std::vector> diff(int periods = 1) const`` - std::vector> - pd_multiindex.h:1449 - :ref:`View ` * - ``MultiIndex difference(const MultiIndex& other, bool sort = true) const`` - MultiIndex - pd_multiindex.h:1473 - :ref:`View ` * - ``MultiIndex shift(int periods = 1, const std::optional& freq = std::nullopt) const`` - MultiIndex - pd_multiindex.h:2692 - :ref:`View ` I/O --- .. list-table:: :widths: 40 20 15 25 :header-rows: 1 * - Signature - Return Type - Location - Example * - ``std::vector> to_flat_index() const`` - std::vector> - pd_multiindex.h:985 - :ref:`View ` * - ``std::vector> to_list() const`` - std::vector> - pd_multiindex.h:2858 - :ref:`View ` * - ``std::vector> to_numpy() const`` - std::vector> - pd_multiindex.h:2866 - :ref:`View ` * - ``std::string to_string() const`` - std::string - pd_multiindex.h:999 - :ref:`View ` * - ``std::vector> tolist() const`` - std::vector> - pd_multiindex.h:2893 - :ref:`View ` Conversion ---------- .. list-table:: :widths: 40 20 15 25 :header-rows: 1 * - Signature - Return Type - Location - Example * - ``MultiIndex astype(const std::string& dtype, bool copy = true) const`` - MultiIndex - pd_multiindex.h:1393 - :ref:`View ` * - ``MultiIndex copy() const`` - MultiIndex - pd_multiindex.h:1264 - :ref:`View ` * - ``MultiIndex infer_objects() const`` - MultiIndex - pd_multiindex.h:2010 - :ref:`View ` * - ``MultiIndex view() const`` - MultiIndex - pd_multiindex.h:3065 - :ref:`View ` Set Operations -------------- .. list-table:: :widths: 40 20 15 25 :header-rows: 1 * - Signature - Return Type - Location - Example * - ``numpy::NDArray duplicated(const std::string& keep = "first") const`` - numpy::NDArray - pd_multiindex.h:1607 - :ref:`View ` * - ``MultiIndex intersection(const MultiIndex& other, bool sort = false) const`` - MultiIndex - pd_multiindex.h:2060 - :ref:`View ` * - ``numpy::NDArray isin(const std::vector>& values, std::optional level = std::nullopt) const`` - numpy::NDArray - pd_multiindex.h:2171 - :ref:`View ` * - ``MultiIndex symmetric_difference(const MultiIndex& other, const std::optional& result_name = std::nullopt, bool sort = true) const`` - MultiIndex - pd_multiindex.h:2789 - :ref:`View ` * - ``MultiIndex union_(const MultiIndex& other, bool sort = true) const`` - MultiIndex - pd_multiindex.h:2939 - :ref:`View ` * - ``MultiIndex unique(std::optional level = std::nullopt) const`` - MultiIndex - pd_multiindex.h:2983 - :ref:`View ` Type Checking ------------- .. list-table:: :widths: 40 20 15 25 :header-rows: 1 * - Signature - Return Type - Location - Example * - ``bool is_(const MultiIndex& other) const`` - bool - pd_multiindex.h:2096 - :ref:`View ` * - ``bool is_boolean() const`` - bool - pd_multiindex.h:2103 - :ref:`View ` * - ``bool is_categorical() const`` - bool - pd_multiindex.h:2110 - :ref:`View ` * - ``bool is_floating() const`` - bool - pd_multiindex.h:2117 - :ref:`View ` * - ``bool is_integer() const`` - bool - pd_multiindex.h:2129 - :ref:`View ` * - ``bool is_interval() const`` - bool - pd_multiindex.h:2142 - :ref:`View ` * - ``bool is_level_ordered(size_t level) const`` - bool - pd_multiindex.h:568 - * - ``bool is_monotonic_decreasing() const`` - bool - pd_multiindex.h:3122 - :ref:`View ` * - ``bool is_monotonic_increasing() const`` - bool - pd_multiindex.h:3108 - :ref:`View ` * - ``bool is_numeric() const`` - bool - pd_multiindex.h:2149 - :ref:`View ` * - ``bool is_object() const`` - bool - pd_multiindex.h:2156 - :ref:`View ` * - ``bool is_unique() const`` - bool - pd_multiindex.h:575 - :ref:`View ` Other Methods ------------- .. list-table:: :widths: 40 20 15 25 :header-rows: 1 * - Signature - Return Type - Location - Example * - ``bool all() const`` - bool - pd_multiindex.h:1276 - :ref:`View ` * - ``bool any() const`` - bool - pd_multiindex.h:1285 - :ref:`View ` * - ``size_t argmax() const`` - size_t - pd_multiindex.h:1292 - :ref:`View ` * - ``size_t argmin() const`` - size_t - pd_multiindex.h:1309 - :ref:`View ` * - ``std::vector> arrays(nlevels)`` - std::vector> - pd_multiindex.h:385 - :ref:`View ` * - ``std::vector> arrays(nlevels)`` - std::vector> - pd_multiindex.h:441 - :ref:`View ` * - ``std::vector> arrays(nlevels())`` - std::vector> - pd_multiindex.h:1243 - :ref:`View ` * - ``std::vector> arrays(nlevels())`` - std::vector> - pd_multiindex.h:2028 - :ref:`View ` * - ``void build_hash_table() const`` - void - pd_multiindex.h:3189 - * - ``const std::vector>& codes() const`` - const std::vector>& - pd_multiindex.h:527 - :ref:`View ` * - ``bool contains(const std::vector& key) const`` - bool - pd_multiindex.h:904 - :ref:`View ` * - ``MultiIndex delete_(size_t loc) const`` - MultiIndex - pd_multiindex.h:1409 - :ref:`View ` * - ``MultiIndex delete_(const std::vector& locs) const`` - MultiIndex - pd_multiindex.h:1430 - :ref:`View ` * - ``std::vector dtypes() const`` - std::vector - pd_multiindex.h:3137 - :ref:`View ` * - ``bool empty() const`` - bool - pd_multiindex.h:513 - :ref:`View ` * - ``void ensure_hash_table() const`` - void - pd_multiindex.h:3203 - * - ``std::pair, MultiIndex> factorize(bool sort = false, bool use_na_sentinel = true) const`` - std::pair, MultiIndex> - pd_multiindex.h:1702 - :ref:`View ` * - ``std::vector format(bool name = true, size_t max_seq_items = 100, bool adjoin = true, std::nullptr_t formatter = nullptr, const std::string& na_rep = "NaN", std::nullptr_t names = nullptr, int space = 1, bool sparsify = true) const`` - std::vector - pd_multiindex.h:1753 - :ref:`View ` * - ``bool has_duplicates() const`` - bool - pd_multiindex.h:593 - :ref:`View ` * - ``bool has_level_categories(size_t level) const`` - bool - pd_multiindex.h:552 - * - ``bool holds_integer() const`` - bool - pd_multiindex.h:1988 - :ref:`View ` * - ``bool identical(const MultiIndex& other) const`` - bool - pd_multiindex.h:2002 - :ref:`View ` * - ``void invalidate_caches() const`` - void - pd_multiindex.h:3237 - * - ``std::vector item() const`` - std::vector - pd_multiindex.h:2237 - :ref:`View ` * - ``static std::string make_tuple_key(const std::vector& values)`` - static std::string - pd_multiindex.h:3212 - * - ``std::string make_tuple_key_at(size_t index) const`` - std::string - pd_multiindex.h:3224 - * - ``size_t memory_usage(bool deep = false) const`` - size_t - pd_multiindex.h:2402 - :ref:`View ` * - ``const std::vector>& names() const`` - const std::vector>& - pd_multiindex.h:534 - :ref:`View ` * - ``size_t nlevels() const`` - size_t - pd_multiindex.h:484 - :ref:`View ` * - ``std::vector> pairs(counts.begin(), counts.end())`` - std::vector> - pd_multiindex.h:3040 - :ref:`View ` * - ``MultiIndex putmask(const numpy::NDArray& mask, const std::vector& value) const`` - MultiIndex - pd_multiindex.h:2474 - :ref:`View ` * - ``std::vector> ravel() const`` - std::vector> - pd_multiindex.h:2507 - :ref:`View ` * - ``MultiIndex remove_unused_levels() const`` - MultiIndex - pd_multiindex.h:835 - :ref:`View ` * - ``MultiIndex repeat(size_t repeats, const std::optional& axis = std::nullopt) const`` - MultiIndex - pd_multiindex.h:2568 - :ref:`View ` * - ``std::string repr() const`` - std::string - pd_multiindex.h:1067 - :ref:`View ` * - ``MultiIndex result(\*this)`` - MultiIndex - pd_multiindex.h:721 - :ref:`View ` * - ``std::vector> result(size())`` - std::vector> - pd_multiindex.h:1450 - :ref:`View ` * - ``std::vector> result_arrays(nlevels())`` - std::vector> - pd_multiindex.h:1476 - * - ``std::vector> result_arrays(nlevels())`` - std::vector> - pd_multiindex.h:1522 - * - ``std::vector> result_arrays(nlevels())`` - std::vector> - pd_multiindex.h:2063 - * - ``std::vector> result_arrays(nlevels())`` - std::vector> - pd_multiindex.h:2263 - * - ``std::vector> result_arrays(nlevels())`` - std::vector> - pd_multiindex.h:2374 - * - ``std::vector> result_arrays(nlevels())`` - std::vector> - pd_multiindex.h:2483 - * - ``std::vector> result_arrays(nlevels())`` - std::vector> - pd_multiindex.h:2571 - * - ``std::vector> result_arrays(nlevels())`` - std::vector> - pd_multiindex.h:2794 - * - ``std::vector> result_arrays(nlevels())`` - std::vector> - pd_multiindex.h:2942 - * - ``std::vector> result_arrays(nlevels())`` - std::vector> - pd_multiindex.h:3084 - * - ``MultiIndex round(int decimals = 0) const`` - MultiIndex - pd_multiindex.h:2593 - :ref:`View ` * - ``MultiIndex set_codes(const std::vector>& codes, std::optional level = std::nullopt, bool verify_integrity = true) const`` - MultiIndex - pd_multiindex.h:2619 - :ref:`View ` * - ``void set_level_categories(size_t level, const std::vector& categories, bool ordered = false)`` - void - pd_multiindex.h:541 - * - ``MultiIndex set_levels(const std::vector>& levels, std::optional level = std::nullopt, bool verify_integrity = true) const`` - MultiIndex - pd_multiindex.h:2655 - :ref:`View ` * - ``size_t size() const`` - size_t - pd_multiindex.h:503 - :ref:`View ` * - ``std::pair slice_indexer( const std::optional>& start = std::nullopt, const std::optional>& end = std::nullopt, size_t step = 1) const`` - std::pair - pd_multiindex.h:2706 - :ref:`View ` * - ``std::pair slice_locs( const std::optional>& start = std::nullopt, const std::optional>& end = std::nullopt, size_t step = 1) const`` - std::pair - pd_multiindex.h:2732 - :ref:`View ` * - ``MultiIndex sort(bool ascending = true) const`` - MultiIndex - pd_multiindex.h:2746 - :ref:`View ` * - ``std::pair> sortlevel( size_t level = 0, bool ascending = true, const std::string& na_position = "last", bool sort_remaining = true) const`` - std::pair> - pd_multiindex.h:1118 - :ref:`View ` * - ``StringMethods str() const`` - StringMethods - pd_multiindex.h:1102 - :ref:`View ` * - ``MultiIndex temp(other)`` - MultiIndex - pd_multiindex.h:156 - * - ``MultiIndex truncate( const std::optional>& before = std::nullopt, const std::optional>& after = std::nullopt) const`` - MultiIndex - pd_multiindex.h:2915 - :ref:`View ` * - ``void verify_integrity_impl() const`` - void - pd_multiindex.h:3172 - Code Examples ------------- The following examples are extracted from the test suite. .. _example-multiindex-multiindex-0: .. dropdown:: MultiIndex (pd_test_3_all.cpp:26015) :class-title: example-dropdown .. code-block:: cpp :linenos: :lineno-start: 26005 :emphasize-lines: 11 // Level 0 (rows): unique rows {0,1,2} = 3 labels if (mi.get_level(0).size() != 3) throw std::runtime_error("Expected 3 row labels, got " + std::to_string(mi.get_level(0).size())); // Level 1 (cols): unique cols {0,1,2} = 3 labels if (mi.get_level(1).size() != 3) throw std::runtime_error("Expected 3 col labels, got " + std::to_string(mi.get_level(1).size())); std::cout << " PASSED" << std::endl; } int pd_test_sparse_coo_main() { try { std::cout << "========= Sparse COO MultiIndex (N1) ==================" << std::endl; pd_test_sparse_coo_non_dense(); pd_test_sparse_coo_dense(); pd_test_sparse_coo_empty(); pd_test_sparse_coo_sorting(); pd_test_sparse_coo_multiindex_levels(); std::cout << "All pd_test_sparse_coo tests passed!" << std::endl; return 0; } catch (const std::exception& e) { std::cout << "FAILED: " << e.what() << std::endl; return 1; .. _example-multiindex-multiindex-1: .. dropdown:: MultiIndex (pd_test_3_all.cpp:26015) :class-title: example-dropdown .. code-block:: cpp :linenos: :lineno-start: 26005 :emphasize-lines: 11 // Level 0 (rows): unique rows {0,1,2} = 3 labels if (mi.get_level(0).size() != 3) throw std::runtime_error("Expected 3 row labels, got " + std::to_string(mi.get_level(0).size())); // Level 1 (cols): unique cols {0,1,2} = 3 labels if (mi.get_level(1).size() != 3) throw std::runtime_error("Expected 3 col labels, got " + std::to_string(mi.get_level(1).size())); std::cout << " PASSED" << std::endl; } int pd_test_sparse_coo_main() { try { std::cout << "========= Sparse COO MultiIndex (N1) ==================" << std::endl; pd_test_sparse_coo_non_dense(); pd_test_sparse_coo_dense(); pd_test_sparse_coo_empty(); pd_test_sparse_coo_sorting(); pd_test_sparse_coo_multiindex_levels(); std::cout << "All pd_test_sparse_coo tests passed!" << std::endl; return 0; } catch (const std::exception& e) { std::cout << "FAILED: " << e.what() << std::endl; return 1; .. _example-multiindex-multiindex-2: .. dropdown:: MultiIndex (pd_test_3_all.cpp:26015) :class-title: example-dropdown .. code-block:: cpp :linenos: :lineno-start: 26005 :emphasize-lines: 11 // Level 0 (rows): unique rows {0,1,2} = 3 labels if (mi.get_level(0).size() != 3) throw std::runtime_error("Expected 3 row labels, got " + std::to_string(mi.get_level(0).size())); // Level 1 (cols): unique cols {0,1,2} = 3 labels if (mi.get_level(1).size() != 3) throw std::runtime_error("Expected 3 col labels, got " + std::to_string(mi.get_level(1).size())); std::cout << " PASSED" << std::endl; } int pd_test_sparse_coo_main() { try { std::cout << "========= Sparse COO MultiIndex (N1) ==================" << std::endl; pd_test_sparse_coo_non_dense(); pd_test_sparse_coo_dense(); pd_test_sparse_coo_empty(); pd_test_sparse_coo_sorting(); pd_test_sparse_coo_multiindex_levels(); std::cout << "All pd_test_sparse_coo tests passed!" << std::endl; return 0; } catch (const std::exception& e) { std::cout << "FAILED: " << e.what() << std::endl; return 1; .. _example-multiindex-from_arrays-3: .. dropdown:: from_arrays (pd_test_1_all.cpp:1994) :class-title: example-dropdown .. code-block:: cpp :linenos: :lineno-start: 1984 :emphasize-lines: 11 // ============================================================================ // Test: from_arrays factory method // ============================================================================ void test_from_arrays() { std::cout << "========= IntervalArray: from_arrays ======================= "; std::vector left_vec = {0, 10, 20}; std::vector right_vec = {5, 15, 25}; auto arr = pandas::IntervalArrayInt64::from_arrays(left_vec, right_vec); if (arr.size() != 3) { std::cout << "[FAIL] : in test_from_arrays() : size" << std::endl; return; } auto interval1 = arr[1]; if (!interval1.has_value() || interval1->first != 10 || interval1->second != 15) { std::cout << "[FAIL] : in test_from_arrays() : interval values" << std::endl; return; .. _example-multiindex-from_arrays-4: .. dropdown:: from_arrays (pd_test_1_all.cpp:1994) :class-title: example-dropdown .. code-block:: cpp :linenos: :lineno-start: 1984 :emphasize-lines: 11 // ============================================================================ // Test: from_arrays factory method // ============================================================================ void test_from_arrays() { std::cout << "========= IntervalArray: from_arrays ======================= "; std::vector left_vec = {0, 10, 20}; std::vector right_vec = {5, 15, 25}; auto arr = pandas::IntervalArrayInt64::from_arrays(left_vec, right_vec); if (arr.size() != 3) { std::cout << "[FAIL] : in test_from_arrays() : size" << std::endl; return; } auto interval1 = arr[1]; if (!interval1.has_value() || interval1->first != 10 || interval1->second != 15) { std::cout << "[FAIL] : in test_from_arrays() : interval values" << std::endl; return; .. _example-multiindex-from_arrays-5: .. dropdown:: from_arrays (pd_test_1_all.cpp:1994) :class-title: example-dropdown .. code-block:: cpp :linenos: :lineno-start: 1984 :emphasize-lines: 11 // ============================================================================ // Test: from_arrays factory method // ============================================================================ void test_from_arrays() { std::cout << "========= IntervalArray: from_arrays ======================= "; std::vector left_vec = {0, 10, 20}; std::vector right_vec = {5, 15, 25}; auto arr = pandas::IntervalArrayInt64::from_arrays(left_vec, right_vec); if (arr.size() != 3) { std::cout << "[FAIL] : in test_from_arrays() : size" << std::endl; return; } auto interval1 = arr[1]; if (!interval1.has_value() || interval1->first != 10 || interval1->second != 15) { std::cout << "[FAIL] : in test_from_arrays() : interval values" << std::endl; return; .. _example-multiindex-from_frame-6: .. dropdown:: from_frame (pd_test_3_all.cpp:9009) :class-title: example-dropdown .. code-block:: cpp :linenos: :lineno-start: 8999 :emphasize-lines: 11 bool neq = mi1.equal_levels(mi3); if (neq) { std::cout << " [FAIL] : in pd_test_3_all_multiindex_equal_levels() : different levels should not be equal" << std::endl; throw std::runtime_error("pd_test_3_all_multiindex_equal_levels failed"); } std::cout << " -> tests passed" << std::endl; } void pd_test_3_all_multiindex_from_frame() { std::cout << "========= MultiIndex.from_frame() ================="; std::vector> columns = { {"a", "a", "b", "b"}, {"x", "y", "x", "y"} }; std::vector> names = {"level0", "level1"}; pandas::MultiIndex mi = pandas::MultiIndex::from_frame(columns, names); if (mi.size() != 4) { .. _example-multiindex-from_mixed_arrays-7: .. dropdown:: from_mixed_arrays (pd_test_1_all.cpp:14830) :class-title: example-dropdown .. code-block:: cpp :linenos: :lineno-start: 14820 :emphasize-lines: 11 void pd_test_multiindex_mixed_types() { std::cout << "========= mixed types (int64 + string) ================ "; std::vector> int_arrays = { {2020, 2020, 2021, 2021} }; std::vector> str_arrays = { {"Q1", "Q2", "Q1", "Q2"} }; pandas::MultiIndex mi = pandas::MultiIndex::from_mixed_arrays( int_arrays, str_arrays, "is", {std::optional("year"), std::optional("quarter")} ); bool passed = true; if (mi.nlevels() != 2) { std::cout << " [FAIL] : nlevels should be 2" << std::endl; passed = false; } .. _example-multiindex-from_product-8: .. dropdown:: from_product (pd_test_1_all.cpp:14246) :class-title: example-dropdown .. code-block:: cpp :linenos: :lineno-start: 14236 :emphasize-lines: 11 } void pd_test_multiindex_from_product() { std::cout << "========= from_product ================================ "; std::vector> iterables = { {"a", "b"}, {"1", "2", "3"} }; pandas::MultiIndex mi = pandas::MultiIndex::from_product(iterables); bool passed = true; // Should have 2*3=6 entries if (mi.size() != 6) { std::cout << " [FAIL] : size should be 6, got " << mi.size() << std::endl; passed = false; } // Check order: (a,1), (a,2), (a,3), (b,1), (b,2), (b,3) .. _example-multiindex-from_product-9: .. dropdown:: from_product (pd_test_1_all.cpp:14246) :class-title: example-dropdown .. code-block:: cpp :linenos: :lineno-start: 14236 :emphasize-lines: 11 } void pd_test_multiindex_from_product() { std::cout << "========= from_product ================================ "; std::vector> iterables = { {"a", "b"}, {"1", "2", "3"} }; pandas::MultiIndex mi = pandas::MultiIndex::from_product(iterables); bool passed = true; // Should have 2*3=6 entries if (mi.size() != 6) { std::cout << " [FAIL] : size should be 6, got " << mi.size() << std::endl; passed = false; } // Check order: (a,1), (a,2), (a,3), (b,1), (b,2), (b,3) .. _example-multiindex-from_tuples-10: .. dropdown:: from_tuples (pd_test_1_all.cpp:2022) :class-title: example-dropdown .. code-block:: cpp :linenos: :lineno-start: 2012 :emphasize-lines: 11 // ============================================================================ void test_from_tuples() { std::cout << "========= IntervalArray: from_tuples ======================= "; std::vector> tuples = { {0.0, 1.5}, {1.5, 3.0}, {3.0, 4.5} }; auto arr = pandas::IntervalArrayFloat64::from_tuples(tuples); if (arr.size() != 3) { std::cout << "[FAIL] : in test_from_tuples() : size" << std::endl; return; } auto interval2 = arr[2]; if (!interval2.has_value() || interval2->first != 3.0 || interval2->second != 4.5) { std::cout << "[FAIL] : in test_from_tuples() : interval values" << std::endl; return; .. _example-multiindex-from_tuples-11: .. dropdown:: from_tuples (pd_test_1_all.cpp:2022) :class-title: example-dropdown .. code-block:: cpp :linenos: :lineno-start: 2012 :emphasize-lines: 11 // ============================================================================ void test_from_tuples() { std::cout << "========= IntervalArray: from_tuples ======================= "; std::vector> tuples = { {0.0, 1.5}, {1.5, 3.0}, {3.0, 4.5} }; auto arr = pandas::IntervalArrayFloat64::from_tuples(tuples); if (arr.size() != 3) { std::cout << "[FAIL] : in test_from_tuples() : size" << std::endl; return; } auto interval2 = arr[2]; if (!interval2.has_value() || interval2->first != 3.0 || interval2->second != 4.5) { std::cout << "[FAIL] : in test_from_tuples() : interval values" << std::endl; return; .. _example-multiindex-get_indexer-12: .. dropdown:: get_indexer (pd_test_1_all.cpp:10332) :class-title: example-dropdown .. code-block:: cpp :linenos: :lineno-start: 10322 :emphasize-lines: 11 void pd_test_extension_index_get_indexer() { std::cout << "========= get_indexer ========================="; pandas::CategoricalArray arr1({"a", "b", "c", "d"}); pandas::CategoricalIndex idx1(arr1); pandas::CategoricalArray arr2({"b", "d", "x"}); pandas::CategoricalIndex idx2(arr2); auto indexer = idx1.get_indexer(idx2); bool passed = (indexer.getSize() == 3 && indexer.getElementAt({0}) == 1 && indexer.getElementAt({1}) == 3 && indexer.getElementAt({2}) == -1); if (!passed) { std::cout << " [FAIL] : in pd_test_extension_index_get_indexer() : get_indexer check failed" << std::endl; throw std::runtime_error("pd_test_extension_index_get_indexer failed"); } .. _example-multiindex-get_indexer_for-13: .. dropdown:: get_indexer_for (pd_test_3_all.cpp:716) :class-title: example-dropdown .. code-block:: cpp :linenos: :lineno-start: 706 :emphasize-lines: 11 // ============================================================================ // Category 6: Index Indexer Methods // ============================================================================ void pd_test_3_all_index_indexers() { std::cout << "========= Index.get_indexer_for/non_unique/slice_indexer() "; std::vector vals = {"a", "b", "c", "d", "e"}; pandas::Index idx(vals); // Test get_indexer_for() std::vector target = {"b", "d", "f"}; // "f" doesn't exist numpy::NDArray indexer = idx.get_indexer_for(target); if (indexer.getSize() != 3) { std::cout << " [FAIL] : in pd_test_3_all_index_indexers() : get_indexer_for size mismatch" << std::endl; throw std::runtime_error("pd_test_3_all_index_indexers failed: get_indexer_for size"); } // "b" is at index 1 if (indexer.getElementAt({0}) != 1) { std::cout << " [FAIL] : in pd_test_3_all_index_indexers() : 'b' should be at index 1" << std::endl; throw std::runtime_error("pd_test_3_all_index_indexers failed: 'b' index"); .. _example-multiindex-get_indexer_non_unique-14: .. dropdown:: get_indexer_non_unique (pd_test_3_all.cpp:739) :class-title: example-dropdown .. code-block:: cpp :linenos: :lineno-start: 729 :emphasize-lines: 11 if (indexer.getElementAt({1}) != 3) { std::cout << " [FAIL] : in pd_test_3_all_index_indexers() : 'd' should be at index 3" << std::endl; throw std::runtime_error("pd_test_3_all_index_indexers failed: 'd' index"); } // "f" doesn't exist -> -1 if (indexer.getElementAt({2}) != -1) { std::cout << " [FAIL] : in pd_test_3_all_index_indexers() : 'f' should be -1" << std::endl; throw std::runtime_error("pd_test_3_all_index_indexers failed: 'f' index"); } // Test get_indexer_non_unique() std::vector target2 = {"a", "c", "z"}; // "z" doesn't exist pandas::Index target_idx(target2); auto [indexer2, missing] = idx.get_indexer_non_unique(target_idx); if (indexer2.getSize() < 2) { std::cout << " [FAIL] : in pd_test_3_all_index_indexers() : get_indexer_non_unique size too small" << std::endl; throw std::runtime_error("pd_test_3_all_index_indexers failed: get_indexer_non_unique size"); } // Test slice_indexer() .. _example-multiindex-get_level-15: .. dropdown:: get_level (pd_test_3_all.cpp:26007) :class-title: example-dropdown .. code-block:: cpp :linenos: :lineno-start: 25997 :emphasize-lines: 11 std::vector rows = {0, 1, 2}; std::vector cols = {1, 2, 0}; auto s = ::pandas::series_from_coo(data, rows, cols, 3, 3, false); if (!s.has_multiindex()) throw std::runtime_error("Expected MultiIndex"); const auto& mi = s.multiindex(); if (mi.nlevels() != 2) throw std::runtime_error("Expected 2 levels, got " + std::to_string(mi.nlevels())); // Level 0 (rows): unique rows {0,1,2} = 3 labels if (mi.get_level(0).size() != 3) throw std::runtime_error("Expected 3 row labels, got " + std::to_string(mi.get_level(0).size())); // Level 1 (cols): unique cols {0,1,2} = 3 labels if (mi.get_level(1).size() != 3) throw std::runtime_error("Expected 3 col labels, got " + std::to_string(mi.get_level(1).size())); std::cout << " PASSED" << std::endl; } int pd_test_sparse_coo_main() { try { std::cout << "========= Sparse COO MultiIndex (N1) ==================" << std::endl; pd_test_sparse_coo_non_dense(); pd_test_sparse_coo_dense(); .. _example-multiindex-get_level_values-16: .. dropdown:: get_level_values (pd_test_3_all.cpp:4524) :class-title: example-dropdown .. code-block:: cpp :linenos: :lineno-start: 4514 :emphasize-lines: 11 } std::cout << " -> tests passed" << std::endl; } void pd_test_3_all_interval_index_get_level_values_droplevel() { std::cout << "========= IntervalIndex.get_level_values/droplevel() "; pandas::IntervalIndex64 idx = pandas::IntervalIndex64::from_breaks({0, 10, 20, 30}); // get_level_values(0) should work pandas::IntervalIndex64 level_vals = idx.get_level_values(0); if (level_vals.size() != idx.size()) { throw std::runtime_error("get_level_values(0) size mismatch"); } // get_level_values(1) should throw bool threw = false; try { idx.get_level_values(1); } catch (const std::out_of_range&) { .. _example-multiindex-get_level_values-17: .. dropdown:: get_level_values (pd_test_3_all.cpp:4524) :class-title: example-dropdown .. code-block:: cpp :linenos: :lineno-start: 4514 :emphasize-lines: 11 } std::cout << " -> tests passed" << std::endl; } void pd_test_3_all_interval_index_get_level_values_droplevel() { std::cout << "========= IntervalIndex.get_level_values/droplevel() "; pandas::IntervalIndex64 idx = pandas::IntervalIndex64::from_breaks({0, 10, 20, 30}); // get_level_values(0) should work pandas::IntervalIndex64 level_vals = idx.get_level_values(0); if (level_vals.size() != idx.size()) { throw std::runtime_error("get_level_values(0) size mismatch"); } // get_level_values(1) should throw bool threw = false; try { idx.get_level_values(1); } catch (const std::out_of_range&) { .. _example-multiindex-get_level_values_str-18: .. dropdown:: get_level_values_str (pd_test_1_all.cpp:14394) :class-title: example-dropdown .. code-block:: cpp :linenos: :lineno-start: 14384 :emphasize-lines: 11 void pd_test_multiindex_get_level_values() { std::cout << "========= get_level_values ============================ "; std::vector> arrays = { {"a", "a", "b"}, {"x", "y", "z"} }; pandas::MultiIndex mi = pandas::MultiIndex::from_arrays(arrays); auto level0_vals = mi.get_level_values_str(0); auto level1_vals = mi.get_level_values_str(1); bool passed = true; if (level0_vals.size() != 3 || level0_vals[0] != "a" || level0_vals[1] != "a" || level0_vals[2] != "b") { std::cout << " [FAIL] : level 0 values mismatch" << std::endl; passed = false; } .. _example-multiindex-get_loc-19: .. dropdown:: get_loc (pd_test_1_all.cpp:10281) :class-title: example-dropdown .. code-block:: cpp :linenos: :lineno-start: 10271 :emphasize-lines: 11 bool passed = (idx.contains("apple") && idx.contains("banana") && !idx.contains("grape")); if (!passed) { std::cout << " [FAIL] : in pd_test_extension_index_contains() : contains check failed" << std::endl; throw std::runtime_error("pd_test_extension_index_contains failed"); } std::cout << " -> tests passed" << std::endl; } void pd_test_extension_index_get_loc_unique() { std::cout << "========= get_loc (unique) ========================="; pandas::CategoricalArray arr({"apple", "banana", "cherry"}); pandas::CategoricalIndex idx(arr); auto loc_apple = idx.get_loc("apple"); auto loc_banana = idx.get_loc("banana"); auto loc_cherry = idx.get_loc("cherry"); bool passed = (std::holds_alternative(loc_apple) && std::get(loc_apple) == 0 && std::get(loc_banana) == 1 && .. _example-multiindex-get_loc_level-20: .. dropdown:: get_loc_level (pd_test_3_all.cpp:9033) :class-title: example-dropdown .. code-block:: cpp :linenos: :lineno-start: 9023 :emphasize-lines: 11 if (mi.nlevels() != 2) { std::cout << " [FAIL] : in pd_test_3_all_multiindex_from_frame() : nlevels mismatch" << std::endl; throw std::runtime_error("pd_test_3_all_multiindex_from_frame failed: nlevels"); } std::cout << " -> tests passed" << std::endl; } void pd_test_3_all_multiindex_get_loc_level() { std::cout << "========= MultiIndex.get_loc_level() =============="; std::vector> arrays = { {"a", "a", "b", "b"}, {"1", "2", "1", "2"} }; pandas::MultiIndex mi = pandas::MultiIndex::from_arrays(arrays); auto [locs, result_mi] = mi.get_loc_level("a", 0, true); if (locs.size() != 2) { .. _example-multiindex-get_loc_string-21: .. dropdown:: get_loc_string (pd_test_3_all.cpp:28108) :class-title: example-dropdown .. code-block:: cpp :linenos: :lineno-start: 28098 :emphasize-lines: 11 vals.push_back(numpy::timedelta64(ns, numpy::DateTimeUnit::Nanosecond)); } return pandas::TimedeltaArray(vals); } void pd_test_getitem_timedelta_str_lookup() { std::cout << " -- pd_test_getitem_timedelta_str_lookup --" << std::endl; int fail = 0; auto tda = ge_make_tda({1 * GE_NS_PER_DAY, 2 * GE_NS_PER_DAY, 3 * GE_NS_PER_DAY}); pandas::TimedeltaIndex tdi(tda); auto pos = tdi.get_loc_string("2 days"); if (!pos.has_value()) { std::cout << " FAIL: '2 days' not found" << std::endl; fail++; } else if (*pos != 1) { std::cout << " FAIL: expected pos=1, got " << *pos << std::endl; fail++; } if (fail == 0) std::cout << " OK" << std::endl; if (fail) throw std::runtime_error("pd_test_getitem_timedelta_str_lookup failed"); } void pd_test_getitem_timedelta_str_not_found() { std::cout << " -- pd_test_getitem_timedelta_str_not_found --" << std::endl; int fail = 0; auto tda = ge_make_tda({1 * GE_NS_PER_DAY}); .. _example-multiindex-get_locs-22: .. dropdown:: get_locs (pd_test_3_all.cpp:9057) :class-title: example-dropdown .. code-block:: cpp :linenos: :lineno-start: 9047 :emphasize-lines: 11 if (locs[0] != 0 || locs[1] != 1) { std::cout << " [FAIL] : in pd_test_3_all_multiindex_get_loc_level() : wrong locations" << std::endl; throw std::runtime_error("pd_test_3_all_multiindex_get_loc_level failed: wrong locs"); } std::cout << " -> tests passed" << std::endl; } void pd_test_3_all_multiindex_get_locs() { std::cout << "========= MultiIndex.get_locs() ==================="; std::vector> arrays = { {"a", "a", "b", "b"}, {"1", "2", "1", "2"} }; pandas::MultiIndex mi = pandas::MultiIndex::from_arrays(arrays); std::vector> seq = {{"a", "1"}, {"b", "2"}}; numpy::NDArray locs = mi.get_locs(seq); .. _example-multiindex-get_name-23: .. dropdown:: get_name (pd_test_5_all.cpp:50005) :class-title: example-dropdown .. code-block:: cpp :linenos: :lineno-start: 49995 :emphasize-lines: 11 std::cout << tag << " [" << c << "]" << " override=" << override_or_empty(df, c) << " dtype=" << series_dtype_or_missing(df, c) << "\n"; } std::cout << tag << " has_multiindex=" << df.has_multiindex() << "\n"; if (df.has_multiindex()) { const auto& mi = df.multiindex(); std::cout << tag << " mi.nlevels=" << mi.nlevels() << " mi.size=" << mi.size() << "\n"; for (size_t i = 0; i < mi.nlevels(); ++i) { auto nm = mi.get_name(i); std::cout << tag << " level[" << i << "] name=" << (nm.has_value() ? *nm : std::string("")) << " level_size=" << mi.get_level(i).size() << " level_dtype=" << mi.get_level(i).dtype_name() << "\n"; } } std::cout << tag << " to_string:\n" << df.to_string() << "\n"; } .. _example-multiindex-get_slice_bound-24: .. dropdown:: get_slice_bound (pd_test_3_all.cpp:3644) :class-title: example-dropdown .. code-block:: cpp :linenos: :lineno-start: 3634 :emphasize-lines: 11 formatted = idx.format(custom_formatter); if (formatted[0] != "val:1") { throw std::runtime_error("custom formatter failed"); } std::cout << " -> tests passed" << std::endl; } void pd_test_3_all_index_get_slice_bound() { std::cout << "========= Index.get_slice_bound() =================="; pandas::Index idx({10, 20, 30, 40, 50}); // Exact match, left side size_t bound = idx.get_slice_bound(30, "left"); if (bound != 2) { throw std::runtime_error("left bound for 30 should be 2"); } // Exact match, right side .. _example-multiindex-get_string-25: .. dropdown:: get_string (pd_test_3_all.cpp:27746) :class-title: example-dropdown .. code-block:: cpp :linenos: :lineno-start: 27736 :emphasize-lines: 11 } } pandas::Series si({10, 20, 30}, "ints"); auto result2 = si.astype("str"); auto* str_s2 = dynamic_cast*>(result2.get()); if (!str_s2) { std::cout << " FAIL: expected Series from int" << std::endl; fail++; } else { if (str_s2->get_string(0) != "10") { std::cout << " FAIL: expected '10', got '" << str_s2->get_string(0) << "'" << std::endl; fail++; } } if (fail == 0) std::cout << " OK" << std::endl; } void pd_test_astype_datetime_to_string() { std::cout << " -- pd_test_astype_datetime_to_string --" << std::endl; .. _example-multiindex-get_tuple_str-26: .. dropdown:: get_tuple_str (pd_test_3_all.cpp:1023) :class-title: example-dropdown .. code-block:: cpp :linenos: :lineno-start: 1013 :emphasize-lines: 11 } for (size_t i = 0; i < bn.size(); ++i) { if (bn[i].value_or("") != nn[i].value_or("")) { std::cout << " [FAIL] level name " << i << " differs: '" << bn[i].value_or("") << "' vs '" << nn[i].value_or("") << "'" << std::endl; throw std::runtime_error("from_arrays brace-init: names divergence"); } } for (size_t i = 0; i < via_brace.size(); ++i) { std::vector tup_brace = via_brace.get_tuple_str(i); std::vector tup_named = via_named.get_tuple_str(i); if (tup_brace != tup_named) { std::cout << " [FAIL] row " << i << " differs" << std::endl; throw std::runtime_error("from_arrays brace-init: content divergence"); } } } // Case C: integer element type - exercises template deduction beyond string { .. _example-multiindex-take-27: .. dropdown:: take (pd_test_1_all.cpp:5903) :class-title: example-dropdown .. code-block:: cpp :linenos: :lineno-start: 5893 :emphasize-lines: 11 // Inherited Operations Tests // ============================================================================ void pd_test_categorical_index_take() { std::cout << "========= inherited take =============================="; pandas::CategoricalArray arr({"a", "b", "c", "d"}); pandas::CategoricalIndex idx(arr); std::vector indices = {0, 2, 3}; pandas::ExtensionIndex taken = idx.take(indices); bool passed = (taken.size() == 3); if (!passed) { std::cout << " [FAIL] : in pd_test_categorical_index_take()" << std::endl; throw std::runtime_error("pd_test_categorical_index_take failed"); } std::cout << " -> tests passed" << std::endl; } .. _example-multiindex-where-28: .. dropdown:: where (pd_test_1_all.cpp:22018) :class-title: example-dropdown .. code-block:: cpp :linenos: :lineno-start: 22008 :emphasize-lines: 11 data["B"] = {5.0, 6.0, 7.0, 8.0}; pandas::DataFrame df(data); // Create condition DataFrame (values > 2) std::map> cond_data; cond_data["A"] = {false, false, true, true}; // 1<=2, 2<=2, 3>2, 4>2 cond_data["B"] = {true, true, true, true}; // all >2 pandas::DataFrame cond(cond_data); // Apply where with replacement value -1 pandas::DataFrame result = df.where(cond, -1.0); // Get column index for A - it's sorted alphabetically in std::map size_t col_a_idx = df.get_column_index("A"); size_t col_b_idx = df.get_column_index("B"); bool passed = true; std::string error_msg; // Check A column values std::string a0 = result.iat(0, col_a_idx) == -1.0 ? "ok" : "fail"; .. _example-multiindex-drop-29: .. dropdown:: drop (pd_test_1_all.cpp:6558) :class-title: example-dropdown .. code-block:: cpp :linenos: :lineno-start: 6548 :emphasize-lines: 11 if (df.ncols() != 2) { std::cout << " [FAIL] : in pd_test_dataframe_manipulation() : pop ncols != 2" << std::endl; throw std::runtime_error("pd_test_dataframe_manipulation failed: pop ncols != 2"); } if (!popped) { std::cout << " [FAIL] : in pd_test_dataframe_manipulation() : popped is null" << std::endl; throw std::runtime_error("pd_test_dataframe_manipulation failed: popped is null"); } // Test drop columns auto dropped = df.drop(std::vector{"B"}, 1); if (dropped.ncols() != 1) { std::cout << " [FAIL] : in pd_test_dataframe_manipulation() : drop ncols != 1" << std::endl; throw std::runtime_error("pd_test_dataframe_manipulation failed: drop ncols != 1"); } // Test rename auto renamed = df.rename_columns(std::map{{"A", "X"}}); if (!renamed.has_column("X")) { std::cout << " [FAIL] : in pd_test_dataframe_manipulation() : rename failed" << std::endl; throw std::runtime_error("pd_test_dataframe_manipulation failed: rename failed"); .. _example-multiindex-drop_duplicates-30: .. dropdown:: drop_duplicates (pd_test_1_all.cpp:6639) :class-title: example-dropdown .. code-block:: cpp :linenos: :lineno-start: 6629 :emphasize-lines: 11 } } // Test drop_duplicates { std::map> dup_data; dup_data["A"] = {1, 1, 2, 2}; dup_data["B"] = {1, 1, 2, 3}; pandas::DataFrame df_dup(dup_data); auto deduped = df_dup.drop_duplicates(); // Rows 0 and 1 are duplicates (A=1, B=1), so should have 3 rows if (deduped.nrows() != 3) { std::cout << " [FAIL] : in pd_test_dataframe_manipulation() : drop_duplicates nrows != 3, got " << deduped.nrows() << std::endl; throw std::runtime_error("pd_test_dataframe_manipulation failed: drop_duplicates"); } } // Test assign { std::map> assign_data; .. _example-multiindex-droplevel-31: .. dropdown:: droplevel (pd_test_1_all.cpp:14428) :class-title: example-dropdown .. code-block:: cpp :linenos: :lineno-start: 14418 :emphasize-lines: 11 void pd_test_multiindex_droplevel() { std::cout << "========= droplevel =================================== "; std::vector> arrays = { {"a", "a", "b"}, {"x", "y", "z"}, {"1", "2", "3"} }; pandas::MultiIndex mi = pandas::MultiIndex::from_arrays(arrays); pandas::MultiIndex dropped = mi.droplevel(1); bool passed = true; if (dropped.nlevels() != 2) { std::cout << " [FAIL] : nlevels should be 2 after drop" << std::endl; passed = false; } // Check remaining levels auto tup = dropped[0]; .. _example-multiindex-dropna-32: .. dropdown:: dropna (pd_test_1_all.cpp:531) :class-title: example-dropdown .. code-block:: cpp :linenos: :lineno-start: 521 :emphasize-lines: 11 } // Test isna array numpy::NDArray na_mask = arr.isna(); if (na_mask.getSize() != 4) { std::cout << " [FAIL] : in pd_test_categorical_array_na_handling() : isna size != 4" << std::endl; throw std::runtime_error("pd_test_categorical_array_na_handling failed: isna size != 4"); } // Test dropna pandas::CategoricalArray dropped = arr.dropna(); if (dropped.size() != 2) { std::cout << " [FAIL] : in pd_test_categorical_array_na_handling() : dropna size != 2" << std::endl; throw std::runtime_error("pd_test_categorical_array_na_handling failed: dropna size != 2"); } // Test fillna (fill with existing category) pandas::CategoricalArray filled = arr.fillna("a"); // 'a' is in categories if (filled.has_na()) { std::cout << " [FAIL] : in pd_test_categorical_array_na_handling() : fillna should have no NA" << std::endl; throw std::runtime_error("pd_test_categorical_array_na_handling failed: fillna should have no NA"); .. _example-multiindex-insert-33: .. dropdown:: insert (pd_test_1_all.cpp:12028) :class-title: example-dropdown .. code-block:: cpp :linenos: :lineno-start: 12018 :emphasize-lines: 11 } std::cout << " -> tests passed" << std::endl; } void pd_test_index_insert_delete() { std::cout << "========= insert and delete ==========================="; pandas::Index idx{1, 2, 4, 5}; auto inserted = idx.insert(2, 3); bool passed = (inserted.size() == 5); passed = passed && (inserted[2] == 3); auto deleted = inserted.delete_(2); passed = passed && (deleted.size() == 4); passed = passed && deleted.equals(idx); if (!passed) { std::cout << " [FAIL] : in pd_test_index_insert_delete() : insert/delete failed" << std::endl; throw std::runtime_error("pd_test_index_insert_delete failed"); .. _example-multiindex-reindex-34: .. dropdown:: reindex (pd_test_1_all.cpp:6708) :class-title: example-dropdown .. code-block:: cpp :linenos: :lineno-start: 6698 :emphasize-lines: 11 } } // Test reindex rows { std::map> data; data["A"] = {1.0, 2.0, 3.0}; pandas::DataFrame df(data); df = df.set_axis({"x", "y", "z"}, 0); auto reindexed = df.reindex({"x", "z", "w"}, 0); if (reindexed.nrows() != 3) { std::cout << " [FAIL] : in pd_test_dataframe_index_ops() : reindex wrong nrows" << std::endl; throw std::runtime_error("pd_test_dataframe_index_ops failed: reindex nrows"); } // 'w' should have NaN std::string val = reindexed["A"].get_value_str(2); if (!std::isnan(std::stod(val))) { std::cout << " [FAIL] : in pd_test_dataframe_index_ops() : missing label should be NaN" << std::endl; throw std::runtime_error("pd_test_dataframe_index_ops failed: reindex NaN"); } .. _example-multiindex-rename-35: .. dropdown:: rename (pd_test_1_all.cpp:5816) :class-title: example-dropdown .. code-block:: cpp :linenos: :lineno-start: 5806 :emphasize-lines: 11 std::cout << " -> tests passed" << std::endl; } void pd_test_categorical_index_rename() { std::cout << "========= rename ======================================"; pandas::CategoricalArray arr({"x", "y"}); pandas::CategoricalIndex idx(arr, "old_name"); pandas::CategoricalIndex renamed = idx.rename("new_name"); bool passed = (renamed.name().has_value() && *renamed.name() == "new_name" && renamed.size() == idx.size() && renamed.categories() == idx.categories()); if (!passed) { std::cout << " [FAIL] : in pd_test_categorical_index_rename()" << std::endl; throw std::runtime_error("pd_test_categorical_index_rename failed"); } std::cout << " -> tests passed" << std::endl; } .. _example-multiindex-reorder_levels-36: .. dropdown:: reorder_levels (pd_test_1_all.cpp:14495) :class-title: example-dropdown .. code-block:: cpp :linenos: :lineno-start: 14485 :emphasize-lines: 11 void pd_test_multiindex_reorder_levels() { std::cout << "========= reorder_levels ============================== "; std::vector> arrays = { {"a", "b"}, {"x", "y"}, {"1", "2"} }; pandas::MultiIndex mi = pandas::MultiIndex::from_arrays(arrays); pandas::MultiIndex reordered = mi.reorder_levels({2, 0, 1}); bool passed = true; auto tup = reordered[0]; if (tup[0] != "1" || tup[1] != "a" || tup[2] != "x") { std::cout << " [FAIL] : reordered tuple should be ('1', 'a', 'x')" << std::endl; passed = false; } if (!passed) { .. _example-multiindex-set_names-37: .. dropdown:: set_names (pd_test_1_all.cpp:14519) :class-title: example-dropdown .. code-block:: cpp :linenos: :lineno-start: 14509 :emphasize-lines: 11 std::cout << "-> tests passed" << std::endl; } void pd_test_multiindex_set_names() { std::cout << "========= set_names =================================== "; std::vector> arrays = {{"a", "b"}, {"x", "y"}}; pandas::MultiIndex mi = pandas::MultiIndex::from_arrays(arrays); std::vector> new_names = {"level_a", "level_b"}; pandas::MultiIndex named = mi.set_names(new_names); bool passed = (named.names()[0] == "level_a" && named.names()[1] == "level_b"); if (!passed) { std::cout << " [FAIL] : names not set correctly" << std::endl; throw std::runtime_error("pd_test_multiindex_set_names failed"); } std::cout << "-> tests passed" << std::endl; } .. _example-multiindex-swaplevel-38: .. dropdown:: swaplevel (pd_test_1_all.cpp:14461) :class-title: example-dropdown .. code-block:: cpp :linenos: :lineno-start: 14451 :emphasize-lines: 11 void pd_test_multiindex_swaplevel() { std::cout << "========= swaplevel =================================== "; std::vector