RosettaCodeData/Task/List-comprehensions/Stata/list-comprehensions.stata
2019-09-12 10:33:56 -07:00

8 lines
176 B
Text

function grid(n,p) {
return(colshape(J(1,p,1::n),1),J(n,1,1::p))
}
n = 20
a = grid(n,n)
a = a,sqrt(a[.,1]:^2+a[.,2]:^2)
a[selectindex(floor(a[.,3]):==a[.,3] :& a[.,3]:<=n),]