Takuya Miyashita

Functionのメモ †

戻り値が複数ある関数を broadcast で実行した時の割り当て †

概要 †

julia で
a, b, c = f(x)
と戻り値 3つ返す f があったとき，これを f.(X) すると戻り値が
Array{Array{T,1}, ndims(X)}
の配列の配列になるんだが，これを
A, B, C (size(A) = size(X))
へとスマートに分配する方法ありますかね…

— ™️ (@hydrocoastjp) August 19, 2020

結果がTupleでも良いなら
A,B,C = zip(f.(X)...)
で行けるかと

— あんちもん2 (@antimon2) August 19, 2020

#Julia言語 https://t.co/pDpXbeWL9j

using StructArrays
f(x) = [sinpi(x), cospi(x)]
X = range(0, 1, length=7)
S, C = f.(X) .|> Tuple |> StructArray |> fieldarrays pic.twitter.com/gwOyCApAtI

— 黒木玄 Gen Kuroki (@genkuroki) August 19, 2020

結局は… †

入力　X が 1次元（ ndims(X) == 1）で　出力を Tuple ではなく Array　にしたい時は

a, b, c = collect.(zip(f.(X)...))

入力　X が 2次元以上　（ ndims(X) > 1） で出力を Array にしたい時は上記の StructArray を使う方法か， zip, collect を利用して

a, b, c = map(x->reshape(x, size(X)), collect.(zip(f.(X)...)))

でも可能．