谷歌番石榴"zip"两个列表



使用Google Guava(Google Commons),有没有办法将两个大小相等的列表合并为一个列表,新列表包含两个输入列表的复合对象?

例:

public class Person {
    public final String name;
    public final int age;
    public Person(String name, int age) {
        this.name = name;
        this.age = age;
    }
    public String toString() {
        return "(" + name + ", " + age + ")";
    }
}

List<String> names = Lists.newArrayList("Alice", "Bob", "Charles");
List<Integer> ages = Lists.newArrayList(42, 27, 31);
List<Person> persons =
    transform with a function that converts (String, Integer) to Person
System.out.println(persons);

将输出:

[(Alice, 42), (Bob, 27), (Charles, 31)]

从番石榴 21 开始,这可以通过 Streams.zip() 实现:

List<Person> persons = Streams.zip(names.stream(), ages.stream(), Person::new)
                              .collect(Collectors.toList());

看起来这目前不在番石榴中,但是一个理想的功能。请参阅此 github 问题,特别是 Iterators.zip()

假装这是一个番石榴方法:

for (int i = 0; i < names.size(); i++) {
    persons.add(new Person(names.get(i), ages.get(i)));
}

你可以参考 underscore-java 库。

Underscore-java是Java的Underscore.js端口,zip方法可以实现目标。

以下是示例代码和输出:

$.zip(Arrays.asList("moe", "larry", "curly"), Arrays.asList("30", "40", "50"));

=> [[萌, 30], [拉里, 40], [卷毛, 50]]

这是一种使用原版 Java 压缩列表的通用方法。缺少元组,我选择使用映射条目列表(如果您不喜欢使用映射条目,请引入额外的类ZipEntry或其他东西)。

public static <T1,T2> List<Map.Entry<T1,T2>> zip(List<T1> zipLeft, List<T2> zipRight) {
    List<Map.Entry<T1,T2>> zipped = new ArrayList<>();
    for (int i = 0; i < zipLeft.size(); i++) {
        zipped.add(new AbstractMap.SimpleEntry<>(zipLeft.get(i), zipRight.get(i)));
    }
    return zipped;
}

要同时支持数组:

@SuppressWarnings("unchecked")
public static <T1,T2> Map.Entry<T1,T2>[] zip(T1[] zipLeft, T2[] zipRight) {
    return zip(asList(zipLeft), asList(zipRight)).toArray(new Map.Entry[] {});
}

为了使其更健壮,请添加对列表大小等的前提条件检查,或引入类似于SQL查询的连接/右连接语义。

这是一个没有明确迭代的版本,但它变得非常丑陋。

List<Person> persons = ImmutableList.copyOf(Iterables.transform(
    ContiguousSet.create(Range.closedOpen(0, names.size()),
        DiscreteDomain.integers()),
    new Function<Integer, Person>() {
      @Override
      public Person(Integer index) {
        return new Person(names.get(index), ages.get(index));
      }
    }));

这实际上并不比显式迭代好多少,您可能需要某种程度的边界检查以确保两个输入确实具有相同的大小。

最新更新