Есть ли в Java эквивалент функции перечисления в Python? - PullRequest
61 голосов
/ 24 августа 2011

В Python функция enumerate позволяет выполнять итерацию по последовательности пар (индекс, значение). Например:

>>> numbers = ["zero", "one", "two"]
>>> for i, s in enumerate(numbers):
...     print i, s
... 
0 zero
1 one
2 two

Есть ли способ сделать это на Java?

Ответы [ 10 ]

50 голосов
/ 24 августа 2011

Для коллекций, которые реализуют интерфейс List, вы можете вызвать метод listIterator(), чтобы получить ListIterator. Итератор имеет (среди прочего) два метода - nextIndex(), чтобы получить индекс; и next(), чтобы получить значение (как и другие итераторы).

Таким образом, Java-эквивалент Python выше может быть:

List<String> numbers = Arrays.asList("zero", "one", "two");
ListIterator<String> it = numbers.listIterator();
while (it.hasNext()) {
    System.out.println(it.nextIndex() + " " + it.next());
}

, который, как и Python, выдает:

0 zero
1 one
2 two
12 голосов
/ 15 января 2016

Я считаю, что это наиболее похоже на подход Python.

Использование

public static void main(String [] args) {
    List<String> strings = Arrays.asList("zero", "one", "two");
    for(EnumeratedItem<String> stringItem : ListUtils.enumerate(strings)) {
        System.out.println(stringItem.index + " " + stringItem.item);
    }
    System.out.println();
    for(EnumeratedItem<String> stringItem : ListUtils.enumerate(strings, 3)) {
        System.out.println(stringItem.index + " " + stringItem.item);
    }
}

Вывод

0 zero
1 one
2 two

3 zero
4 one
5 two

Функции

  • Работает с любой итерацией
  • Не создает копию списка в памяти (подходит для больших списков)
  • Поддерживает собственный для каждого синтаксиса
  • Принимает параметр запуска, который можетдобавить в индекс

Реализация

import java.util.Iterator;

public class ListUtils {

    public static class EnumeratedItem<T> {
        public T item;
        public int index;

        private EnumeratedItem(T item, int index) {
            this.item = item;
            this.index = index;
        }
    }

    private static class ListEnumerator<T> implements Iterable<EnumeratedItem<T>> {

        private Iterable<T> target;
        private int start;

        public ListEnumerator(Iterable<T> target, int start) {
            this.target = target;
            this.start = start;
        }

        @Override
        public Iterator<EnumeratedItem<T>> iterator() {
            final Iterator<T> targetIterator = target.iterator();
            return new Iterator<EnumeratedItem<T>>() {

                int index = start;

                @Override
                public boolean hasNext() {
                    return targetIterator.hasNext();
                }

                @Override
                public EnumeratedItem<T> next() {
                    EnumeratedItem<T> nextIndexedItem = new EnumeratedItem<T>(targetIterator.next(), index);
                    index++;
                    return nextIndexedItem;
                }

            };
        }

    }

    public static <T> Iterable<EnumeratedItem<T>> enumerate(Iterable<T> iterable, int start) {
        return new ListEnumerator<T>(iterable, start);
    }

    public static <T> Iterable<EnumeratedItem<T>> enumerate(Iterable<T> iterable) {
        return enumerate(iterable, 0);
    }

}
8 голосов
/ 24 августа 2011

Строго говоря, нет, так как функция enumerate () в Python возвращает список кортежей, а в Java кортежей не существует.

Если, однако, все, что вас интересует, это печать из индекса и значения, тогда вы можете последовать совету Ричарда Фирна и использовать nextIndex () и next () на итераторе.

Также обратите внимание, что enumerate () может быть определен с использованием более общей функции zip () (с использованием синтаксиса Python):

mylist = list("abcd")
zip(range(len(mylist)), mylist)

дает [(0, 'a'), (1, 'b'), (2, 'c'), (3, 'd')]

Если вы определите свой собственный класс Tuple (см. Использование пар или 2-кортежей в Java в качестве отправной точки), то вы, безусловно, можете легко написать свою собственную функцию zip () в Java, чтобы использовать это (используя класс Tuple, определенный в ссылке):

public static <X,Y> List<Tuple<X,Y>> zip(List<X> list_a, List<Y> list_b) {
    Iterator<X> xiter = list_a.iterator();
    Iterator<Y> yiter = list_b.iterator();

    List<Tuple<X,Y>> result = new LinkedList<Tuple<X,Y>>();

    while (xiter.hasNext() && yiter.hasNext()) {
        result.add(new Tuple<X,Y>(xiter.next(), yiter.next()));
    }

    return result;
}

И если у вас есть zip (), реализация enumerate () становится тривиальной.

Редактировать: медленный день на работе, чтобы закончить его:

public static <X> List<Tuple<Integer,X>> enumerate (List<X> list_in) {
    List<Integer> nums = new ArrayList<Integer>(list_in.size());
    for (int x = 0; x < list_in.size(); x++) { 
        nums.add(Integer.valueOf(x));
    }

    return zip (nums, list_in);
}

Редактировать 2: как указано в комментариях к этому вопросу, это не совсем эквивалентно. Хотя он генерирует те же значения, что и перечисление Python, он делает это не так, как перечисление Python. Таким образом, для больших коллекций этот подход может быть весьма запретительным.

4 голосов
/ 25 января 2013

Согласно документации по Python ( здесь ), это самое близкое, что вы можете получить с Java, и оно больше не многословно:

String[] numbers = {"zero", "one", "two"}
for (int i = 0; i < numbers.length; i++) // Note that length is a property of an array, not a function (hence the lack of () )
    System.out.println(i + " " + numbers[i]);
}

Если вам нужно использовать List класс ...

List<String> numbers = Arrays.asList("zero", "one", "two");
for (int i = 0; i < numbers.size(); i++) {
    System.out.println(i + " " + numbers.get(i));
}

* ПРИМЕЧАНИЕ: если вам нужно изменить список при его обходе, вам нужно будет использовать объект Iterator, поскольку он имеет возможность изменять список без вызова ConcurrentModificationException.

2 голосов
/ 24 августа 2011
List<String> list = { "foo", "bar", "foobar"};
int i = 0;
for (String str : list){
     System.out.println(i++ + str );
}
1 голос
/ 13 октября 2017

Просто и понятно

public static <T> void enumerate(Iterable<T> iterable, java.util.function.ObjIntConsumer<T> consumer) {
    int i = 0;
    for(T object : iterable) {
        consumer.accept(object, i);
        i++;
    }
}

Пример использования:

void testEnumerate() {
    List<String> strings = Arrays.asList("foo", "bar", "baz");
    enumerate(strings, (str, i) -> {
        System.out.println(String.format("Index:%d String:%s", i, str));
    });
}
1 голос
/ 10 июля 2016

Теперь с Java 8s Stream API вместе с небольшой библиотекой ProtonPack, обеспечивающей StreamUtils, это может быть легко достигнуто.

В первом примере используется та же запись для каждой записи, что и дляв вопросе:

Stream<String> numbers = Arrays.stream("zero one two".split(" "));
List<Indexed<String>> indexedNumbers = StreamUtils.zipWithIndex(numbers)
                                                  .collect(Collectors.toList());
for (Indexed<String> indexed : indexedNumbers) {
    System.out.println(indexed.getIndex() + " " + indexed.getValue());
}

Выше, хотя и не обеспечивает ленивую оценку, как в Python.Для этого вы должны использовать forEach() Stream API метод:

Stream<String> numbers = Arrays.stream("zero one two".split(" "));
StreamUtils.zipWithIndex(numbers)
        .forEach(n -> System.out.println(n.getIndex() + " " + n.getValue()));

Ленивая оценка может быть проверена следующим бесконечным потоком:

Stream<Integer> infStream = Stream.iterate(0, i -> i++);
StreamUtils.zipWithIndex(infStream)
        .limit(196)
        .forEach(n -> System.out.println(n.getIndex() + " " + n.getValue()));
1 голос
/ 17 июня 2015

Комбинируя генерики с анонимными интерфейсами, вы можете по существу создать фабричный метод для обработки перечисления.Обратный вызов перечислителя скрывает беспорядок итератора под ним.

import java.util.Arrays;
import java.util.List;
import java.util.ListIterator;

public class ListUtils2 {
    public static interface Enumerator<T> {
        void execute(int index, T value);
    };

    public static final <T> void enumerate(final List<T> list,
            final Enumerator<T> enumerator) {
        for (ListIterator<T> it = list.listIterator(); it.hasNext();) {
            enumerator.execute(it.nextIndex(), it.next());
        }
    }

    public static final void enumerate(final String[] arr,
            final Enumerator<String> enumerator) {
        enumerate(Arrays.asList(arr), enumerator);
    }

    public static void main(String[] args) {
        String[] names = { "John", "Paul", "George", "Ringo" };

        enumerate(names, new Enumerator<String>() {
            @Override
            public void execute(int index, String value) {
                System.out.printf("[%d] %s%n", index, value);
            }
        });
    }
}

Результат

[0] John
[1] Paul
[2] George
[3] Ringo

Расширенные мысли

Карта, уменьшение, фильтр

Я сделал еще один шаг вперед и создал функции отображения, сокращения и фильтрации на основе этой концепции.

Обе зависимости Google Guava и Apache для общих коллекцийвключить аналогичный функционал.Вы можете проверить их по своему желанию.

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.ListIterator;

public class ListUtils {
    // =========================================================================
    // Enumerate
    // =========================================================================
    public static abstract interface Enumerator<T> {
        void execute(int index, T value, List<T> list);
    };

    public static final <T> void enumerate(final List<T> list,
            final Enumerator<T> enumerator) {
        for (ListIterator<T> it = list.listIterator(); it.hasNext();) {
            enumerator.execute(it.nextIndex(), it.next(), list);
        }
    }

    // =========================================================================
    // Map
    // =========================================================================
    public static interface Transformer<T, U> {
        U execute(int index, T value, List<T> list);
    };

    public static final <T, U> List<U> transform(final List<T> list,
            final Transformer<T, U> transformer) {
        List<U> result = new ArrayList<U>();
        for (ListIterator<T> it = list.listIterator(); it.hasNext();) {
            result.add(transformer.execute(it.nextIndex(), it.next(), list));
        }
        return result;
    }

    // =========================================================================
    // Reduce
    // =========================================================================
    public static interface Reducer<T, U> {
        U execute(int index, T value, U result, List<T> list);
    };

    public static final <T, U> U reduce(final List<T> list,
            final Reducer<T, U> enumerator, U result) {
        for (ListIterator<T> it = list.listIterator(); it.hasNext();) {
            result = enumerator.execute(it.nextIndex(), it.next(), result, list);
        }
        return result;
    }

    // =========================================================================
    // Filter
    // =========================================================================
    public static interface Predicate<T> {
        boolean execute(int index, T value, List<T> list);
    };

    public static final <T> List<T> filter(final List<T> list,
            final Predicate<T> predicate) {
        List<T> result = new ArrayList<T>();
        for (ListIterator<T> it = list.listIterator(); it.hasNext();) {
            int index = it.nextIndex();
            T value = it.next();
            if (predicate.execute(index, value, list)) {
                result.add(value);
            }
        }
        return result;
    }

    // =========================================================================
    // Predefined Methods
    // =========================================================================
    // Enumerate
    public static <T> String printTuples(List<T> list) {
        StringBuffer buff = new StringBuffer();

        enumerate(list, new Enumerator<T>() {
            @Override
            public void execute(int index, T value, List<T> list) {
                buff.append('(').append(index).append(", ")
                    .append(value).append(')');
                if (index < list.size() - 1) {
                    buff.append(", ");
                }
            }
        });

        return buff.toString();
    }

    // Map
    public static List<String> intToHex(List<Integer> list) {
        return transform(list, new Transformer<Integer, String>() {
            @Override
            public String execute(int index, Integer value, List<Integer> list) {
                return String.format("0x%02X", value);
            }
        });
    }

    // Reduce
    public static Integer sum(List<Integer> list) {
        return reduce(list, new Reducer<Integer, Integer>() {
            @Override
            public Integer execute(int index, Integer value, Integer result,
                    List<Integer> list) {
                return result + value;
            }
        }, 0);
    }

    // Filter
    public static List<Integer> evenNumbers(List<Integer> list) {
        return filter(list, new Predicate<Integer>() {
            @Override
            public boolean execute(int index, Integer value, List<Integer> list) {
                return value % 2 == 0;
            }
        });
    }

    // =========================================================================
    // Driver
    // =========================================================================
    public static void main(String[] args) {
        List<Integer> numbers = Arrays.asList(8, 6, 7, 5, 3, 0, 9);

        // Enumerate
        System.out.printf("%-10s: %s%n", "Enumerate", printTuples(numbers));

        // Map
        System.out.printf("%-10s: %s%n", "Map", intToHex(numbers));

        // Reduce
        System.out.printf("%-10s: %d%n", "Reduce", sum(numbers));

        // Filter
        System.out.printf("%-10s: %s%n", "Filter", evenNumbers(numbers));
    }
}
1 голос
/ 16 января 2015

Я думаю, что это должна быть Java-функциональность, которая больше всего напоминает Python, «перечисляя», хотя это довольно сложно и неэффективно. По сути, просто сопоставьте индексы списка с его элементами, используя ListIterator или Collector:

List<String> list = new LinkedList<>(Arrays.asList("one", "two", "three", "four"));
Map<Integer, String> enumeration = new Map<>();
ListIterator iter = list.listIterator();
while(iter.hasNext){
    map.put(iter.nextIndex(), iter.next());
}

или используя лямбда-выражение:

Set<Integer, String> enumeration = IntStream.range(0, list.size()).boxed.collect(Collectors.toMap(index -> index, index -> list.get(index)));

тогда вы можете использовать его с расширенным циклом for:

for (Map.Entry<Integer, String> entry : enumeration.entrySet){
    System.out.println(entry.getKey() + "\t" + entry.getValue());
}
1 голос
/ 24 августа 2011

Нет. Может быть, есть некоторые библиотеки для поддержки такой функциональности. Но если вы прибегаете к стандартным библиотекам, это ваша работа, чтобы сосчитать.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...