feat<map> add ValuesByKeys for get values array in order [issue - 467]

README.md

@@ -130,6 +130,7 @@ Supported helpers for maps:
 - [PickBy](#pickby)
 - [PickByKeys](#pickbykeys)
 - [PickByValues](#pickbyvalues)
+- [ValuesByKeys](#ValuesByKeys)
 - [OmitBy](#omitby)
 - [OmitByKeys](#omitbykeys)
 - [OmitByValues](#omitbyvalues)
@@ -1120,6 +1121,18 @@ m := lo.PickByValues(map[string]int{"foo": 1, "bar": 2, "baz": 3}, []int{1, 3})
 
 [[play](https://go.dev/play/p/1zdzSvbfsJc)]
 
+### ValuesByKeys
+
+Returns an array of values in the same order as the given keys.
+
+```go
+
+m := lo.ValuesByKeys(map[string]int{"foo": 1, "bar": 2, "baz": 3}, []string{"baz", "foo", "bar"})
+// []int{3，1，2}
+```
+
+[[play](https://go.dev/play/p/PmmPDo1AqWl)]
+
 ### OmitBy
 
 Returns same map type filtered by given predicate.

map.go

@@ -76,6 +76,18 @@ func PickByValues[K comparable, V comparable, Map ~map[K]V](in Map, values []V)
 	return r
 }
 
+// ValuesByKeys returns an array of values in the same order as the given keys.
+// Play: https://go.dev/play/p/PmmPDo1AqWl
+func ValuesByKeys[K comparable, V any](in map[K]V, keys []K) []V {
+	out := make([]V, 0, len(keys))
+	for i := range keys {
+		if v, ok := in[keys[i]]; ok {
+			out = append(out, v)
+		}
+	}
+	return out
+}
+
 // OmitBy returns same map type filtered by given predicate.
 // Play: https://go.dev/play/p/EtBsR43bdsd
 func OmitBy[K comparable, V any, Map ~map[K]V](in Map, predicate func(key K, value V) bool) Map {

map_example_test.go

@@ -66,6 +66,15 @@ func ExamplePickByValues() {
 	// Output: 2 1 3
 }
 
+func ExampleValuesByKeys() {
+	kv := map[string]int{"foo": 1, "bar": 2, "baz": 3}
+
+	result := ValuesByKeys(kv, []string{"baz", "foo", "bar"})
+
+	fmt.Printf("%v %v %v", len(result), result[0], result[1])
+	// Output: 3 3 1
+}
+
 func ExampleOmitBy() {
 	kv := map[string]int{"foo": 1, "bar": 2, "baz": 3}
 

map_test.go

@@ -95,6 +95,19 @@ func TestPickByValues(t *testing.T) {
 	is.IsType(after, before, "type preserved")
 }
 
+func TestValuesByKeys(t *testing.T) {
+	t.Parallel()
+	is := assert.New(t)
+
+	result1 := ValuesByKeys(map[string]int{"foo": 1, "bar": 2, "baz": 3}, []string{"baz", "foo", "bar"})
+	result2 := ValuesByKeys(map[string]int{"": 0, "foobar": 6, "baz": 3}, []string{"baz", "foobar"})
+
+	is.Equal(len(result1), 3)
+	is.Equal(len(result2), 2)
+	is.ElementsMatch(result1, []int{3, 1, 2})
+	is.ElementsMatch(result2, []int{3, 6})
+}
+
 func TestOmitBy(t *testing.T) {
 	t.Parallel()
 	is := assert.New(t)