logo

줄리아에서 컴프리헨션하는 법 📂줄리아

줄리아에서 컴프리헨션하는 법

코드

줄리아에서도 파이썬처럼 컴프리헨션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