Javaで２つのリストを同時にループ処理したい

Question

Pythonでいうところのzip関数使って２つのリストをloop処理することは、
Javaでどう書けばよいですか？
そもそもできなさそう。。。
従来forではなく拡張forで書きたいです。

for i, j in zip(list1, list2):
    pass

Comments

[jimbe]

なぜわざわざ拡張 for を使うのでしょうか。

Answer 1

余計なコードが必要になると思います。

import java.util.*;

class Value<E1,E2> {
  final E1 i;
  final E2 j;
  Value(E1 i, E2 j) {
    this.i = i;
    this.j = j;
  }
  static <E1,E2> Collection<Value<E1,E2>> zip(List<E1> iList, List<E2> jList) {
    int size = Math.min(iList.size(), jList.size());
    List<Value<E1,E2>> list = new ArrayList<>(size);
    for(int i=0; i<size; i++) list.add(new Value<E1,E2>(iList.get(i), jList.get(i)));
    return list;
  }
}

public class Main {
  public static void main(String[] args) throws Exception {
    List<Integer> list1 = Arrays.asList(1,2,3,4);
    List<String> list2 = Arrays.asList("A","B","C");

    for(Value<Integer,String> v : Value.zip(list1, list2)) {
      System.out.println("i="+v.i+", j="+v.j);
    }
  }
}

i=1, j=A
i=2, j=B
i=3, j=C

nukasa さんが仰られるように、普通に for で書けば以下で済むことです。

import java.util.*;

public class Main {
  public static void main(String[] args) throws Exception {
    List<Integer> list1 = Arrays.asList(1,2,3,4);
    List<String> list2 = Arrays.asList("A","B","C");

    int size = Math.min(list1.size(), list2.size());
    for(int i=0; i<size; i++) {
      System.out.println("i="+list1.get(i)+", j="+list2.get(i));
    }
  }
}

Comments

[xebme]

jimbeさんのおっしゃるように仕掛けが大きくなります。拡張forで利用するにはIterableを実装するクラスのインスタンスが必要です。tuple相当のクラスも定義しなければなりません。

[viennaCoffee]

回答してくださった方々、ありがとうございます。
素直なforで実装しようと思います。
やろうと思えば実装できることもわかり、とても勉強になりました。

Answer 2

Javaではできないです。
list1とlist2の要素数が同じだとわかっていれば、list1でループを回し、ループの中でlist2から要素を一つずつ取り出すことで、同等のことはできますが。
要素数が違う場合は、無理くりやるなら、こんなコードになるでしょう。

try {
    var it2 = list2.iterator();
    for (Foo i : list1) {
        j = it2.next();
        ...
    }
} catch (NoSuchElementException e) {}

Javaでの一般的なやり方としては、「従来for」でlist1とlist2の要素数の少ない方分だけループし、list1、list2双方からgetメソッドで値を取得するようにしますね。

Answer 3

while

やってみると、Javaならiteratorを2つ使うwhileが自然です。

		var a = List.of(1,2,3,4);
		var b = List.of("x","y","z","a","b");
		var l1 = a.iterator();
		var l2 = b.iterator();
		while (l1.hasNext() && l2.hasNext())
			System.out.println(l1.next() + " " + l2.next());

拡張for

pythonicに拡張forでやれ。つぎのようにIterableを使います。タプルも定義。

		var a = List.of(1,2,3,4);
		var b = List.of("x","y","z","a","b");
		for (var e : new Zip<>(a,b))
			System.out.println(e.x() + " " + e.y());

Iterable

拡張forは内部でIterableを実装するクラスを使います。しかし、上のwhileをIterableでラップして大げさにしているだけです。

import java.util.Iterator;
import java.util.List;

record Tuple<T,U>(T x, U y){}

class Zip<T,U> implements Iterable<Tuple<T,U>> {
	
	final Iterator<T> l1;
	final Iterator<U> l2;
	
	public Zip(List<T> list1,List<U> list2) {
		l1 = list1.iterator();
		l2 = list2.iterator();
	}
	
	@Override 
	public Iterator<Tuple<T,U>> iterator() {
			return new Iterator<>() {
				@Override
				public boolean hasNext() {
					return l1.hasNext() && l2.hasNext();
				}
				@Override
				public Tuple<T,U> next() {
					if (!hasNext()) {
						throw new RuntimeException();
					}
					return new Tuple<>(l1.next(),l2.next());
				}
			};
		}
}

Comments

[xebme]

ふりかえると、zipに関して恥ずかしい回答をたくさんしています。これで終わりにします。