RosettaCodeData/Task/Apply-a-callback-to-an-array/Stata/apply-a-callback-to-an-array.stata

function map(f,a) {
	nr = rows(a)
	nc = cols(a)
	b = J(nr,nc,.)
	for (i=1;i<=nr;i++) {
		for (j=1;j<=nc;j++) b[i,j] = (*f)(a[i,j])
	}
	return(b)
}

function maps(f,a) {
	nr = rows(a)
	nc = cols(a)
	b = J(nr,nc,"")
	for (i=1;i<=nr;i++) {
		for (j=1;j<=nc;j++) b[i,j] = (*f)(a[i,j])
	}
	return(b)
}

function square(x) {
	return(x*x)
}