Map iterator function in GO

iter_func_map.go

package main

import (
	"fmt"
	"iter"
	"math"
	"strings"
)

// Map : A -> B
func Map[Slice ~[]A, A any, B any](s Slice, f func(A) B) iter.Seq2[int, B] {
	return func(yield func(int, B) bool) {
		for i, v := range s {
			if !yield(i, f(v)) {
				return
			}
		}
	}
}

func main() {
	for i, v := range Map([]float64{1, 2, 3, 4, 5}, math.IsNaN) {
		fmt.Println(i, v)
	}
	for i, v := range Map([]string{"hello", "world"}, strings.ToUpper) {
		fmt.Println(i, v)
	}
}