Merge pull request #774 from ChrisRackauckas/complex

Recipe for handing arrays of complex values
This commit is contained in:
Michael Krabbe Borregaard 2017-04-18 20:17:35 +02:00 committed by GitHub
commit a3b4b9c8b2

View File

@ -801,12 +801,19 @@ datetimeformatter(dt) = string(convert(DateTime, dt))
# -------------------------------------------------
# Complex Numbers
@userplot ComplexPlot
@recipe function f(cp::ComplexPlot)
xguide --> "Real Part"
yguide --> "Imaginary Part"
seriestype --> :scatter
real(cp.args[1]), imag(cp.args[1])
@recipe function f{T<:Number}(A::Array{Complex{T}})
xguide --> "Re(x)"
yguide --> "Im(x)"
real.(A), imag.(A)
end
# Splits a complex matrix to its real and complex parts
# Reals defaults solid, imaginary defaults dashed
# Label defaults are changed to match the real-imaginary reference / indexing
@recipe function f{T<:Real,T2}(x::AbstractArray{T},y::Array{Complex{T2}})
ylabel --> "Re(y)"
zlabel --> "Im(y)"
x,real.(y),imag.(y)
end