mirror of
https://github.com/MillironX/beefblup.git
synced 2024-11-14 11:33:10 +00:00
Make column names more robust
This commit is contained in:
parent
91fc553c4c
commit
4b66ad8b1f
1 changed files with 21 additions and 0 deletions
|
@ -58,6 +58,9 @@ function beefblup(path::String, savepath::String, h2::Float64)
|
||||||
# Import data from a suitable spreadsheet
|
# Import data from a suitable spreadsheet
|
||||||
data = DataFrame(CSV.File(path))
|
data = DataFrame(CSV.File(path))
|
||||||
|
|
||||||
|
# Make sure the data is in the proper format
|
||||||
|
renamecolstospec!(data)
|
||||||
|
|
||||||
# Sort the array by date
|
# Sort the array by date
|
||||||
sort!(data, :birthdate)
|
sort!(data, :birthdate)
|
||||||
|
|
||||||
|
@ -293,5 +296,23 @@ function additiverelationshipmatrix(id::AbstractVector, damid::AbstractVector, s
|
||||||
return A
|
return A
|
||||||
end
|
end
|
||||||
|
|
||||||
|
"""
|
||||||
|
renamecolstospec(::DataFrame)
|
||||||
|
|
||||||
|
Renames the first four columns of the beefblup data sheet so that they can be referred to by
|
||||||
|
name instead of by column index, regardless of user input.
|
||||||
|
"""
|
||||||
|
function renamecolstospec!(df::DataFrame)
|
||||||
|
# Pull out the fixed-effect and observation name
|
||||||
|
othernames = propertynames(df)[5:end]
|
||||||
|
|
||||||
|
# Put specification column names and user-defined names together
|
||||||
|
allnames = cat([:id, :birthdate, :dam, :sire], othernames, dims=1)
|
||||||
|
|
||||||
|
# Rename in the DataFrame
|
||||||
|
rename!(df, allnames, makeunique=true)
|
||||||
|
return df
|
||||||
|
end
|
||||||
|
|
||||||
|
|
||||||
end
|
end
|
||||||
|
|
Loading…
Reference in a new issue