add ODESolution (and SciMLEnsembleSolution) to DataFrame support (con…

…ditional on 1.9)

Project.toml

@@ -29,6 +29,12 @@ SymbolicIndexingInterface = "2efcf032-c050-4f8e-a9bb-153293bab1f5"
 Tables = "bd369af6-aec1-5ad0-b16a-f7cc5008161c"
 TruncatedStacktraces = "781d530d-4396-4725-bb49-402e4bee1e77"
 
+[weakdeps]
+DataFrames = "a93c6f00-e57d-5684-b7b6-d8193f3e46c0"
+
+[extensions]
+SciMLBaseDataFramesExt = "DataFrames"
+
 [compat]
 ADTypes = "0.1.3"
 ArrayInterface = "6, 7"

ext/SciMLBaseDataFramesExt.jl

@@ -0,0 +1,28 @@
+module SciMLBaseDataFramesExt
+using SciMLBase, DataFrames
+
+function DataFrames.DataFrame(sol::EnsembleSolution, idxs::AbstractVector)
+ @assert allequal(getproperty.(sol.u, :t)) "solutions must have shared timesteps"
+ data = sol[:, idxs]
+ i_max, j_max, _ = size(data)
+ v = ["t"=>sol[1].t]
+ for (i, s) in enumerate(sol.u)
+ for idx in idxs
+ push!(v, string("sol ", i, ": ", idx)=>s[idx])
+ end
+ end
+ DataFrame(v)
+end
+
+function DataFrames.DataFrame(sol::ODESolution, idxs::AbstractVector)
+ @assert allequal(getproperty.(sol.u, :t)) "solutions must have shared timesteps"
+ data = sol[:, idxs]
+ i_max, j_max, _ = size(data)
+ v = ["t"=>sol[1].t]
+ for idx in idxs
+ push!(v, string("sol ", i, ": ", idx)=>s[idx])
+ end
+ DataFrame(v)
+end
+
+end