Julia - lấy phần thực của mảng phức tạp

Question

Tôi cần chia biến số z::Array{Complex128,1} thành hai mảng cho các phần thực và phức tạp. Một cách làm điều này là để làm cho biến mới ::Array{Float64,1} và điền họ phần tử bằng phần tử:

for i = 1:size(z)[1] 
    ri[i] = z[i].re 
    ii[i] = z[i].im 
end 

Có cách nào để làm điều này mà không liên quan đến việc sao chép dữ liệu, như bằng cách nào đó thao tác những bước tiến và hiệu số của z?

Answer 1

Chỉnh sửa: Phiên bản gốc đã sử dụng thao tác reshape không cần thiết. Như @ DDF chỉ ra trong các ý kiến, đó là không cần thiết. Câu trả lời đã được sửa đổi kể từ đó.

Trong trường hợp phổ biến mà sao chép không phải là vấn đề, chỉ cần thực hiện real(z) và imag(z) (đổi tên real.(z) và imag.(z) trong v0.6). Tôi bao gồm điều này để giúp độc giả tương lai có vấn đề tương tự, nhưng những người có thể không quan tâm đến việc sao chép.

Như bạn đề xuất, bạn có thể thao tác các bước tiến của z để tránh sao chép dữ liệu. Đơn giản chỉ cần

zfl = reinterpret(Float64, z) 
zre = @view zfl[1:2:end-1] 
zim = @view zfl[2:2:end] 

Kết hợp, chúng tôi quan sát thấy không có sao chép dữ liệu (phân bổ là do chế độ xem mảng được phân bổ và tối thiểu).

julia> z = Vector{Complex128}(100000); 

julia> function reimvec(z) 
      zfl = reinterpret(Float64, z) 
      zre = @view zfl[1:2:end-1] 
      zim = @view zfl[2:2:end] 
      zre, zim 
     end 
reimvec (generic function with 1 method) 

julia> @time reimvec(z); 
    0.000005 seconds (9 allocations: 400 bytes) 

Như chúng ta có thể thấy, đằng sau hậu trường, một mảng như thế được strided:

julia> strides(reimvec(z)[1]) 
(2,)