简体   繁体   中英

Golang equivalent for Python's list comprehension

I am playing with Go but I am having a very hard time doing things that are very simple in other languages.

I'd like to reproduce a similar syntax:

array = [a for a in anotherArray  if (some condition)]

What is an elegant way to do it in Go? I'd really like to simplify my code especially when using function on array. For example:

min = min(abs(a[i], b[j]) for i in range(n)
                          for j in range(i, n))

Many thanks

Interestingly enough, Rob Pike just proposed (18 hours ago) the library filter which does a bit what you want:

See for instance Choose()

// Choose takes a slice of type []T and a function of type func(T) bool. (If
// the input conditions are not satisfied, Choose panics.) It returns a newly
// allocated slice containing only those elements of the input slice that
// satisfy the function.

Tested here :

func TestChoose(t *testing.T) {
    a := []int{1, 2, 3, 4, 5, 6, 7, 8, 9}
    expect := []int{2, 4, 6, 8}
    result := Choose(a, isEven)

As twotwotwo points out in the comments , the GoDoc for this library states:

Package filter contains utility functions for filtering slices through the distributed application of a filter function.

The package is an experiment to see how easy it is to write such things in Go. It is easy, but for loops are just as easy and more efficient .

You should not use this package.

This caveat is reflected in the document " Summary of Go Generics Discussions ", section " Functional Code ":

These are the usual higher-order functions such as map , reduce ( fold ), filter , zip etc.

Cases :
typesafe data transformations: map , fold , zip

Pros for using generics :
Concise way to express data transformations.

Cons for using generics :
The fastest solution needs to take into account when and which order to apply those transformations, and how much data is generated at each step.
It is harder to read for beginners.

Alternative solutions :

use for loops and usual language constructs .

If what you are looking is indeed python list comprehension, there is no such syntactic equivalent in go AFAIK.

The way to do it is to create a function that take a slice and a function (to test the condition) and return a new slice.

EDIT: Well looks like there is already such a feature in Go. cf VonC

The technical post webpages of this site follow the CC BY-SA 4.0 protocol. If you need to reprint, please indicate the site URL or the original address.Any question please contact:yoyou2525@163.com.

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