示例#1
0
 def test_map_str(self, data, categories, ordered):
     # GH 31202 - override base class since we want to maintain categorical/ordered
     index = CategoricalIndex(data, categories=categories, ordered=ordered)
     result = index.map(str)
     expected = CategoricalIndex(map(str, data),
                                 categories=map(str, categories),
                                 ordered=ordered)
     tm.assert_index_equal(result, expected)
示例#2
0
    def test_map_with_dict_or_series(self):
        orig_values = ["a", "B", 1, "a"]
        new_values = ["one", 2, 3.0, "one"]
        cur_index = CategoricalIndex(orig_values, name="XXX")
        expected = CategoricalIndex(new_values,
                                    name="XXX",
                                    categories=[3.0, 2, "one"])

        mapper = Series(new_values[:-1], index=orig_values[:-1])
        result = cur_index.map(mapper)
        # Order of categories in result can be different
        tm.assert_index_equal(result, expected)

        mapper = {o: n for o, n in zip(orig_values[:-1], new_values[:-1])}
        result = cur_index.map(mapper)
        # Order of categories in result can be different
        tm.assert_index_equal(result, expected)
示例#3
0
    def test_map(self):
        ci = CategoricalIndex(list("ABABC"),
                              categories=list("CBA"),
                              ordered=True)
        result = ci.map(lambda x: x.lower())
        exp = CategoricalIndex(list("ababc"),
                               categories=list("cba"),
                               ordered=True)
        tm.assert_index_equal(result, exp)

        ci = CategoricalIndex(list("ABABC"),
                              categories=list("BAC"),
                              ordered=False,
                              name="XXX")
        result = ci.map(lambda x: x.lower())
        exp = CategoricalIndex(list("ababc"),
                               categories=list("bac"),
                               ordered=False,
                               name="XXX")
        tm.assert_index_equal(result, exp)

        # GH 12766: Return an index not an array
        tm.assert_index_equal(
            ci.map(lambda x: 1),
            Index(np.array([1] * 5, dtype=np.int64), name="XXX"))

        # change categories dtype
        ci = CategoricalIndex(list("ABABC"),
                              categories=list("BAC"),
                              ordered=False)

        def f(x):
            return {"A": 10, "B": 20, "C": 30}.get(x)

        result = ci.map(f)
        exp = CategoricalIndex([10, 20, 10, 20, 30],
                               categories=[20, 10, 30],
                               ordered=False)
        tm.assert_index_equal(result, exp)

        result = ci.map(Series([10, 20, 30], index=["A", "B", "C"]))
        tm.assert_index_equal(result, exp)

        result = ci.map({"A": 10, "B": 20, "C": 30})
        tm.assert_index_equal(result, exp)