줄리아에서 컴프리헨션하는 법
코드
줄리아에서도 파이썬처럼 컴프리헨션comprehension이 가능하다. 컴프리헨션이란 배열을 생성하는 방법 중 하나로 조건식 자체를 배열에 기입하는 것을 말한다.
가령 $0$부터 $9$까지의 정수를 순서대로 원소로 갖는 배열을 정의하고 싶다면, for 루프 자체를 배열에 넣어주면 된다.
julia> [i for i ∈ 0:9]
10-element Vector{Int64}:
0
1
2
3
4
5
6
7
8
9
julia> [i*ones(8) for i ∈ 1:9]
9-element Vector{Vector{Float64}}:
[1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0]
[2.0, 2.0, 2.0, 2.0, 2.0, 2.0, 2.0, 2.0]
[3.0, 3.0, 3.0, 3.0, 3.0, 3.0, 3.0, 3.0]
[4.0, 4.0, 4.0, 4.0, 4.0, 4.0, 4.0, 4.0]
[5.0, 5.0, 5.0, 5.0, 5.0, 5.0, 5.0, 5.0]
[6.0, 6.0, 6.0, 6.0, 6.0, 6.0, 6.0, 6.0]
[7.0, 7.0, 7.0, 7.0, 7.0, 7.0, 7.0, 7.0]
[8.0, 8.0, 8.0, 8.0, 8.0, 8.0, 8.0, 8.0]
[9.0, 9.0, 9.0, 9.0, 9.0, 9.0, 9.0, 9.0]
julia> stack([i*ones(8) for i ∈ 1:9], dims=2)
8×9 Matrix{Float64}:
1.0 2.0 3.0 4.0 5.0 6.0 7.0 8.0 9.0
1.0 2.0 3.0 4.0 5.0 6.0 7.0 8.0 9.0
1.0 2.0 3.0 4.0 5.0 6.0 7.0 8.0 9.0
1.0 2.0 3.0 4.0 5.0 6.0 7.0 8.0 9.0
1.0 2.0 3.0 4.0 5.0 6.0 7.0 8.0 9.0
1.0 2.0 3.0 4.0 5.0 6.0 7.0 8.0 9.0
1.0 2.0 3.0 4.0 5.0 6.0 7.0 8.0 9.0
1.0 2.0 3.0 4.0 5.0 6.0 7.0 8.0 9.0
이중 for문도 가능하다.
julia> X = 1:4
1:4
julia> Y = [-1, 0, 1]
3-element Vector{Int64}:
-1
0
1
julia> [x*y for x=X, y=Y]
4×3 Matrix{Int64}:
-1 0 1
-2 0 2
-3 0 3
-4 0 4
환경
- OS: Windows11
- Version: Julia 1.10.0