Plotly: Как установить стиль и цвет для стрелок, используя add_annotations?

Question

Я хочу представить двумерные векторы с корнем в начале координат и графиком в R. Кроме того, я хочу раскрасить вектор на основе категориальной переменной. Проблема в том, что я могу создать строки с цветовой кодировкой, но без стрелки:

library(plotly)
library(dplyr)

v <- c(1, 1)
b1 <- c(1, 0)
b2 <- c(0, 1)
df <- data.frame(
  x = c(v[1], b1[1], b2[1]),
  y = c(v[2], b1[2], b2[2]),
  is_basis = c(FALSE, TRUE, TRUE)
)
df %>% 
  plot_ly(x = ~x, y = ~y, color = ~is_basis) %>%
  add_segments(xend = ~x, yend = ~y, x = 0, y = 0, colors = c("red","black"))

Или с помощью стрелки, но без цветовой кодировки:

df %>% 
  plot_ly(x = ~x, y = ~y, color = ~is_basis) %>%
  add_annotations(x = ~x, y = ~y, showarrow = TRUE, text = "", ax = 0, ay = 0,
                  axref = "x", ayref = "y", xref = "x", yref = "y")

Так , мой вопрос: я могу добавить стрелку с add_segments? Или можно стилизовать и раскрасить стрелки, созданные с помощью add_annotations?

Answer 1

Вы можете выбрать стрелку и установить цвет стрелки с помощью аргументов arrowhead и arrowcolor в функции add_annotations(). Если вы установите axref = "x" и ayref = "y", вы также убедитесь, что строки начинаются с ori go, если для ax и ay установлено значение 0. Вот пример использования подмножества встроенных наборов данных mtcars:

Полный код:

library(plotly)
library(dplyr)

data <- mtcars[which(mtcars$am == 1 & mtcars$gear == 4),]
#data <- data %>% filter(row.names(data) %in% c("Honda Civic", "Fiat X1-9", "Datsun 710"))
#row.names(data) <- c("Honda Civic", "Fiat X1-9", "Datsun 710")

p <- plot_ly(data, x = ~wt, y = ~mpg, type = 'scatter', mode = 'markers',
        marker = list(size = 2)) #%>%

p <- p %>%
  add_annotations(x = data["Honda Civic","wt"],
                  y = data["Honda Civic","mpg"],
                  text = "",
                  xref = "x",
                  yref = "y",
                  showarrow = TRUE,
                  arrowcolor='blue',
                  arrowhead = 1,
                  arrowsize = 2,
                  ax = 0,
                  ay = 0,
                  axref="x",
                  ayref='y',
                  font = list(color = '#264E86',
                              family = 'sans serif',
                              size = 14))

p <- p %>%
  add_annotations(x = data["Fiat X1-9","wt"],
                  y = data["Fiat X1-9","mpg"],
                  text = "",
                  xref = "x",
                  yref = "y",
                  showarrow = TRUE,
                  arrowcolor='green',
                  arrowhead = 2,
                  arrowsize = 2,
                  ax = 0,
                  ay = 0,
                  axref="x",
                  ayref='y',
                  font = list(color = '#264E86',
                              family = 'sans serif',
                              size = 14))

p <- p %>%
  add_annotations(x = data["Datsun 710","wt"],
                  y = data["Datsun 710","mpg"],
                  text = "",
                  xref = "x",
                  yref = "y",
                  showarrow = TRUE,
                  arrowcolor='red',
                  arrowhead = 3,
                  arrowsize = 2,
                  ax = 0,
                  ay = 0,
                  axref="x",
                  ayref='y',
                  font = list(color = '#264E86',
                              family = 'sans serif',
                              size = 14))

p <- p %>%
  add_annotations(x = data["Fiat 128","wt"],
                  y = data["Fiat 128","mpg"],
                  text = "",
                  xref = "x",
                  yref = "y",
                  showarrow = TRUE,
                  arrowcolor='black',
                  arrowhead = 4,
                  arrowsize = 2,
                  ax = 0,
                  ay = 0,
                  axref="x",
                  ayref='y',
                  font = list(color = '#264E86',
                              family = 'sans serif',
                              size = 14))

p <- p %>%
  add_annotations(x = data["Mazda RX4 Wag","wt"],
                  y = data["Mazda RX4 Wag","mpg"],
                  text = "",
                  xref = "x",
                  yref = "y",
                  showarrow = TRUE,
                  arrowcolor='purple',
                  arrowhead = 5,
                  arrowsize = 2,
                  ax = 0,
                  ay = 0,
                  axref="x",
                  ayref='y',
                  font = list(color = '#264E86',
                              family = 'sans serif',
                              size = 14))


p <- p %>%
  add_annotations(x = data["Volvo 142E","wt"],
                  y = data["Volvo 142E","mpg"],
                  text = "",
                  xref = "x",
                  yref = "y",
                  showarrow = TRUE,
                  arrowcolor='grey',
                  arrowhead = 6,
                  arrowsize = 2,
                  ax = 0,
                  ay = 0,
                  axref="x",
                  ayref='y',
                  font = list(color = '#264E86',
                              family = 'sans serif',
                              size = 14))

p <- p %>%
  add_annotations(x = data["Toyota Corolla","wt"],
                  y = data["Toyota Corolla","mpg"],
                  text = "",
                  xref = "x",
                  yref = "y",
                  showarrow = TRUE,
                  arrowcolor='aquamarine',
                  arrowhead = 7,
                  arrowsize = 2,
                  ax = 0,
                  ay = 0,
                  axref="x",
                  ayref='y',
                  font = list(color = '#264E86',
                              family = 'sans serif',
                              size = 14))

p