Как отформатировать ось X жестко закодированной функции построения пакета SPEI в R? - PullRequest
2 голосов
/ 03 мая 2019

Я использую пакет SPEI вместе с его выборочными ежемесячными данными за 32 года.Я хочу изменить метки оси X, чтобы они отражали годы, а не цифры.Тем не менее, жестко закодированная функция построения графиков не позволит мне сделать это.Я устал извлекать данные SPEI$fitted и пытался повторить тот же график с помощью ggplot, но безуспешно.Вот пример кода

install.packages("SPEI")
library(SPEI)
data("wichita")
wichita$PET=hargreaves(Tmin=wichita$TMIN, Tmax = wichita$TMAX, lat = 37.64)
wichita$BAL=wichita$PRCP - wichita$PET
SPEI_12=spei(wichita[,"BAL"],12)
plot.spei(SPEI_12, main = 12-Month SPEI)

Любая помощь будет оценена.Я хочу создать график как прикрепленный. enter image description here

1 Ответ

3 голосов
/ 03 мая 2019

Я не совсем понял функцию plot.spei, поэтому использовал ggplot2.

По сути, я построил фрейм данных с ts установленных значений и создал цвет / заливку условие для положительных (pos) или отрицательных (neg) значений.

library(zoo)
library(tidyverse)
DF <- zoo::fortify.zoo(SPEI_12$fitted)
DF <- DF %>% 
  dplyr::select(-Index) %>% 
  dplyr::mutate(Period = zoo::as.yearmon(paste(wichita$YEAR, wichita$MONTH), "%Y %m")) %>% 
  na.omit() %>% 
  dplyr::mutate(sign = ifelse(ET0_har >= 0, "pos", "neg"))

ggplot2::ggplot(DF) +
  geom_bar(aes(x = Period, y = ET0_har, col = sign, fill = sign),
            show.legend = F, stat = "identity") +
  scale_color_manual(values = c("pos" = "darkblue", "neg" = "red")) +
  scale_fill_manual(values = c("pos"  = "darkblue", "neg" = "red")) +
  scale_y_continuous(limits = c(-3, 3), 
                     breaks = -3:3) +
  ylab("SPEI") + ggtitle("12-Month SPEI") +
  theme_bw() + theme(plot.title = element_text(hjust = 0.5))

enter image description here


Редактировать : дополнительная идея.

DF2 <- DF %>% 
  tidyr::spread(sign, ET0_har) %>% 
  replace(is.na(.), 0)

ggplot2::ggplot(DF2) + 
  geom_area(aes(x = Period, y = pos), fill = "blue", col = "black") +
  geom_area(aes(x = Period, y = neg), fill = "red",  col = "black") +
  scale_y_continuous(limits = c(-3, 3), 
                     breaks = -3:3) +
  ylab("SPEI") + ggtitle("12-Month SPEI") +
  theme_bw() + theme(plot.title = element_text(hjust = 0.5))

enter image description here

...