我知道如何“转换”一个简单的Java列表从Y -> Z,即:

List<String> x;
List<Integer> y = x.stream()
        .map(s -> Integer.parseInt(s))
        .collect(Collectors.toList());

现在我想对Map做基本相同的事情,即:

INPUT:
{
  "key1" -> "41",    // "41" and "42"
  "key2" -> "42"      // are Strings
}

OUTPUT:
{
  "key1" -> 41,      // 41 and 42
  "key2" -> 42       // are Integers
}

解决方案不应局限于String -> Integer。就像上面的List示例一样,我想调用任何方法(或构造函数)。


当前回答

像这样的一般解

public static <X, Y, Z> Map<X, Z> transform(Map<X, Y> input,
        Function<Y, Z> function) {
    return input
            .entrySet()
            .stream()
            .collect(
                    Collectors.toMap((entry) -> entry.getKey(),
                            (entry) -> function.apply(entry.getValue())));
}

例子

Map<String, String> input = new HashMap<String, String>();
input.put("string1", "42");
input.put("string2", "41");
Map<String, Integer> output = transform(input,
            (val) -> Integer.parseInt(val));

其他回答

我的StreamEx库增强了标准流API,提供了一个EntryStream类,更适合转换地图:

Map<String, Integer> output = EntryStream.of(input).mapValues(Integer::valueOf).toMap();

下面是Sotirios Delimanolis的答案的一些变化,它以(+1)开始非常好。考虑以下几点:

static <X, Y, Z> Map<X, Z> transform(Map<? extends X, ? extends Y> input,
                                     Function<Y, Z> function) {
    return input.keySet().stream()
        .collect(Collectors.toMap(Function.identity(),
                                  key -> function.apply(input.get(key))));
}

这里有几点。首先是在泛型中使用通配符;这使得函数在某种程度上更加灵活。通配符是必要的,例如,如果你想要输出映射有一个键是输入映射的键的超类:

Map<String, String> input = new HashMap<String, String>();
input.put("string1", "42");
input.put("string2", "41");
Map<CharSequence, Integer> output = transform(input, Integer::parseInt);

(这里也有一个映射值的例子,但这真的是人为的,我承认为Y设置有界通配符只在边缘情况下有用。)

第二点是,我没有在输入映射的entrySet上运行流,而是在keySet上运行流。我认为这使得代码更简洁,代价是必须从map条目中获取值,而不是从map条目中获取值。顺便说一句,我最初有key -> key作为toMap()的第一个参数,由于某种原因,这失败了,导致类型推断错误。将其更改为(X键)->键,就像Function.identity()一样。

另一种说法如下:

static <X, Y, Z> Map<X, Z> transform1(Map<? extends X, ? extends Y> input,
                                      Function<Y, Z> function) {
    Map<X, Z> result = new HashMap<>();
    input.forEach((k, v) -> result.put(k, function.apply(v)));
    return result;
}

它使用Map.forEach()而不是流。我认为,这甚至更简单,因为它省去了收集器,而收集器与地图一起使用有些笨拙。原因是map . foreach()将键和值作为单独的参数提供,而流只有一个值——您必须选择是使用键还是使用映射条目作为该值。缺点是,这种方法缺乏其他方法丰富、流畅的优点。: -)

像这样的一般解

public static <X, Y, Z> Map<X, Z> transform(Map<X, Y> input,
        Function<Y, Z> function) {
    return input
            .entrySet()
            .stream()
            .collect(
                    Collectors.toMap((entry) -> entry.getKey(),
                            (entry) -> function.apply(entry.getValue())));
}

例子

Map<String, String> input = new HashMap<String, String>();
input.put("string1", "42");
input.put("string2", "41");
Map<String, Integer> output = transform(input,
            (val) -> Integer.parseInt(val));

如果你不介意使用第三方库,我的cyclops-react库有所有JDK集合类型的扩展,包括Map。我们可以直接使用'map'操作符转换map(默认情况下map作用于map中的值)。

   MapX<String,Integer> y = MapX.fromMap(HashMaps.of("hello","1"))
                                .map(Integer::parseInt);

Bimap可用于同时转换键和值

  MapX<String,Integer> y = MapX.fromMap(HashMaps.of("hello","1"))
                               .bimap(this::newKey,Integer::parseInt);

它一定要100%的功能性和流畅性吗?如果没有,那么下面这个怎么样,它是最简短的:

Map<String, Integer> output = new HashMap<>();
input.forEach((k, v) -> output.put(k, Integer.valueOf(v));

(如果你能忍受将流和副作用结合在一起的羞耻和内疚的话)