繁体   English   中英

Java 中类似 Python 的列表理解

[英]Python-like list comprehension in Java

由于 Java 不允许将方法作为参数传递,那么在 Java 中您使用什么技巧来实现 Python 之类的列表理解?

我有一个字符串列表(ArrayList)。 我需要使用函数来转换每个元素,以便获得另一个列表。 我有几个函数将一个字符串作为输入并返回另一个字符串作为输出。 我如何创建一个通用方法,该方法可以将列表和函数作为参数,以便我可以获取处理每个元素的列表。 从字面上看这是不可能的,但是我应该使用什么技巧?

另一种选择是为每个较小的字符串处理函数编写一个新函数,它只是循环遍历整个列表,这有点不太酷。

在 Java 8 中,您可以使用方法引用:

List<String> list = ...;
list.replaceAll(String::toUpperCase);

或者,如果您想创建一个新的列表实例:

List<String> upper = list.stream().map(String::toUpperCase).collect(Collectors.toList());

基本上,您创建一个 Function 接口:

public interface Func<In, Out> {
    public Out apply(In in);
}

然后将匿名子类传递给您的方法。

您的方法可以将函数应用到每个元素就地:

public static <T> void applyToListInPlace(List<T> list, Func<T, T> f) {
    ListIterator<T> itr = list.listIterator();
    while (itr.hasNext()) {
        T output = f.apply(itr.next());
        itr.set(output);
    }
}
// ...
List<String> myList = ...;
applyToListInPlace(myList, new Func<String, String>() {
    public String apply(String in) {
        return in.toLowerCase();
    }
});

或创建一个新List (基本上创建从输入列表到输出列表的映射):

public static <In, Out> List<Out> map(List<In> in, Func<In, Out> f) {
    List<Out> out = new ArrayList<Out>(in.size());
    for (In inObj : in) {
        out.add(f.apply(inObj));
    }
    return out;
}
// ...
List<String> myList = ...;
List<String> lowerCased = map(myList, new Func<String, String>() {
    public String apply(String in) {
        return in.toLowerCase();
    }
});

哪一个更可取取决于您的用例。 如果您的列表非常大,就地解决方案可能是唯一可行的解​​决方案; 如果您希望将许多不同的函数应用于同一个原始列表以制作许多衍生列表,您将需要map版本。

Google Collections 库有许多类用于在比普通 Java 支持的更高级别上以功能方式(过滤器、映射、折叠等)处理集合和迭代器。 它定义了 Function 和 Predicate 接口以及使用它们来处理集合的方法,这样您就不必这样做了。 它还具有方便的功能,使处理 Java 泛型变得不那么困难。

我还使用Hamcrest ** 过滤集合。

这两个库很容易与适配器类结合。


** 利益声明:我共同编写了 Hamcrest

Apache Commons CollectionsUtil.transform(Collection, Transformer)是另一种选择。

我正在构建这个项目来用 Java 编写列表理解,现在是https://github.com/farolfo/list-comprehension-in-java 中的概念证明

例子

// { x | x E {1,2,3,4} ^ x is even }
// gives {2,4}

Predicate<Integer> even = x -> x % 2 == 0;

List<Integer> evens = new ListComprehension<Integer>()
    .suchThat(x -> {
        x.belongsTo(Arrays.asList(1, 2, 3, 4));
        x.is(even);
    });
// evens = {2,4};

如果我们想以某种方式转换输出表达式,例如

// { x * 2 | x E {1,2,3,4} ^ x is even }
// gives {4,8}

List<Integer> duplicated = new ListComprehension<Integer>()
    .giveMeAll((Integer x) -> x * 2)
    .suchThat(x -> {
        x.belongsTo(Arrays.asList(1, 2, 3, 4));
        x.is(even);
    });
// duplicated = {4,8}

您可以对函数使用 lambda,如下所示:

class Comprehension<T> {
    /**
    *in: List int
    *func: Function to do to each entry
    */
    public List<T> comp(List<T> in, Function<T, T> func) {
        List<T> out = new ArrayList<T>();
        for(T o: in) {
            out.add(func.apply(o));
        }
        return out;
    }
}

用法:

List<String> stuff = new ArrayList<String>();
stuff.add("a");
stuff.add("b");
stuff.add("c");
stuff.add("d");
stuff.add("cheese");
List<String> newStuff = new Comprehension<String>().comp(stuff, (a) -> { //The <String> tells the comprehension to return an ArrayList<String>
    a.equals("a")? "1": 
            (a.equals("b")? "2": 
                (a.equals("c")? "3":
                    (a.equals("d")? "4": a
    )))
});

将返回:

["1", "2", "3", "4", "cheese"]
import java.util.Arrays;

class Soft{
    public static void main(String[] args){
        int[] nums=range(9, 12);
        System.out.println(Arrays.toString(nums));
    }
    static int[] range(int low, int high){
        int[] a=new int[high-low];
        for(int i=0,j=low;i<high-low;i++,j++){
            a[i]=j;
        }
        return a;

    }
}

希望,我能帮助你:)

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM