List comprehensions: Difference between revisions
m
no edit summary
mNo edit summary |
mNo edit summary |
||
Line 2,006:
#(12 16 20)"
</lang>
Stata does no have list comprehensions, but the Mata matrix language helps simplify this task.
<lang stata>function grid(n,p) {
return(colshape(J(p,1,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]),]</lang>
'''Output'''
<pre> 1 2 3
+----------------+
1 | 3 4 5 |
2 | 4 3 5 |
3 | 5 12 13 |
4 | 6 8 10 |
5 | 8 6 10 |
6 | 8 15 17 |
7 | 9 12 15 |
8 | 12 5 13 |
9 | 12 9 15 |
10 | 12 16 20 |
11 | 15 8 17 |
12 | 15 20 25 |
13 | 16 12 20 |
14 | 20 15 25 |
+----------------+</pre>
=={{header|SuperCollider}}==
Line 2,167 ⟶ 2,200:
{{omit from|Go}}
{{omit from|Maxima}}
▲{{omit from|Stata}}
| |||