propagate NaN value in median

JuliaStats · Mar 13, 2024 · cc11ea9 · cc11ea9
1 parent 68869af
commit cc11ea9
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/Statistics.jl b/src/Statistics.jl
@@ -814,7 +814,8 @@ Like [`median`](@ref), but may overwrite the input vector.
 function median!(v::AbstractVector)
     isempty(v) && throw(ArgumentError("median of an empty array is undefined, $(repr(v))"))
     eltype(v)>:Missing && any(ismissing, v) && return missing
-    any(x -> x isa Number && isnan(x), v) && return convert(eltype(v), NaN)
+    nanix = findfirst(x -> x isa Number && isnan(x), v)
+    isnothing(nanix) || return v[nanix]
     inds = axes(v, 1)
     n = length(inds)
     mid = div(first(inds)+last(inds),2)