R/traits.R

In drhutools: Political Science Academic Research Gears

#' @name traits
#' @title Calculate psychological traits based on
#'
#' @param survey Psychological survey data as a `data.frame`.
#'
#' @details
#' The current version can calculate the TOSCA-3SC scores and Grit-O score.
#'
#' @return A data frame with trait scores for each observations
#'
#' @references
#' Duckworth, Angela L., Christopher Peterson, Michael D. Matthews, and Dennis R. Kelly. 2007. 
#' “Grit: Perseverance and Passion for Long-Term Goals.” Journal of Personality and Social Psychology 
#' 92(6): 1087–1101. doi:10.1037/0022-3514.92.6.1087.
#'
#' Tangney, June P. 1990. “Assessing Individual Differences in Proneness to Shame and Guilt: Development 
#' of the Self-Conscious Affect and Attribution Inventory.” Journal of Personality and Social Psychology 
#' 59(1): 102–11. doi:10.1037/0022-3514.59.1.102.
#'
#' @examples
#' column_names <- c("Q3|R3", "Q3|R4", "Q4|R3", "Q4|R4", "Q5|R5", "Q5|R6", "Q6|R3", "Q6|R4", "Q7|R3", 
#' "Q7|R4", "Q8|R5", "Q8|R6", "Q9|R5", "Q9|R6", "Q10|R5", "Q10|R6", "Q11|R5", "Q11|R6", "Q12|R3", 
#' "Q12|R4", "Q13|R3", "Q13|R4", "Q14|1", "Q15|1", "Q16|1", "Q17|1", "Q18|1", "Q19|1", "Q20|1", 
#' "Q21|1", "Q22|1", "Q23|1", "Q24|1", "Q25|1")
#'
#' toy_data <- data.frame(matrix(sample(1:5, 10 * length(column_names), replace = TRUE), 
#' ncol = length(column_names)))
#'
#' names(toy_data) <- column_names
#'
#' traits(toy_data)
#'
#' @export

traits <- function(survey) {
  # Calculate score_shame using columns that match ".*R(3|5)"
  score_shame <- survey[, grep(".*R(3|5)", names(survey))] |>
    rowSums(na.rm = TRUE)

  # Calculate score_guilt using columns that match ".*R(4|6)"
  score_guilt <- survey[, grep(".*R(4|6)", names(survey))] |>
    rowSums(na.rm = TRUE)

  # Calculate score_grit using columns that end with ".1"
  score_grit <- survey[, grep(".*\\|1$", names(survey))] |>
    rowMeans(na.rm = TRUE)

  # Create a data frame with the results
  df_result <- data.frame(score_shame = score_shame, score_guilt = score_guilt, score_grit = score_grit)

  return(df_result)
}