在 R 中添加颜色和气泡大小图例

Question

可能很简单。

我有一个 xy 数据集，我想使用R 的plotly 进行绘图。以下是数据：

set.seed(1)
df <- data.frame(x=1:10,y=runif(10,1,10),group=c(rep("A",9),"B"),group.size=as.integer(runif(10,1,10)))

我想用df$group 为数据着色，并让点的大小跟随df$group.size（即气泡图）。另外，我想同时添加这两个图例。

这是我天真的尝试：

require(plotly)  
require(dplyr)  

    main.plot <-
 plot_ly(type='scatter',mode="markers",color=~df$group,x=~df$x,y=~df$y,size=~df$group.size,marker=list(sizeref=0.1,sizemode="area",opacity=0.5),data=df,showlegend=T) %>%
     layout(title="Title",xaxis=list(title="X",zeroline=F),yaxis=list(title="Y",zeroline=F))

结果如下：

不幸的是，这个图例搞砸了，至少我希望它是这样的：每个组的一个点具有相同的大小但颜色不同。

然后为我关注this的group.size添加图例，也得到了aocall的回答：

legend.plot <- plot_ly() %>% add_markers(x = 1, y = unique(df$group.size),
                                size = unique(df$group.size),
                                showlegend = T, 
                                marker = list(sizeref=0.1,sizemode="area")) %>%
   layout(title="TITLE",xaxis = list(zeroline=F,showline=F,showticklabels=F,showgrid=F),
          yaxis=list(showgrid=F))

结果如下：

我的问题是图例包含我的数据中不存在的值。

然后我使用subplot 组合它们：

subplot(legend.plot, main.plot, widths = c(0.1, 0.9))

我明白了：

图例标题被删除的地方

所以我会提供一些帮助。

Answer 1

根据更新的请求：

注意legend.plot 的变化（将值映射到整数序列，然后手动更改轴刻度文本），以及使用注释获取图例标题。正如in this answer 解释的那样，无论使用多少个子图，都只能使用一个标题。

图例上的圆圈似乎对应于每条迹线的最小点大小。因此，我在 (12, 12) 处添加了一个点，并限制了轴的范围以确保它不显示。

titleX 和titleY 控制轴标签的显示，如here 所述。

set.seed(1)
df <- data.frame(x=1:10,y=runif(10,1,10),group=c(rep("A",9),"B"),group.size=as.integer(runif(10,1,10)))
require(plotly)  
require(dplyr)  


## Take unique values before adding dummy value
unique_vals <- unique(df$group.size)
df <- rbind(c(12, 12, "B", 1), df)
df[c(1, 2, 4)] <- lapply(df[c(1, 2, 4)], as.numeric)


main.plot <-
  plot_ly(type='scatter',
    mode="markers",
    color=~df$group,
    x=~df$x,
    y=~df$y,
    size=~df$group.size,
    marker=list(
        sizeref=0.1,
        sizemode="area",
        opacity=0.5),
        data=df,
        showlegend=T) %>%
  layout(title="Title",
    xaxis=list(title="X",zeroline=F, range=c(0, 11)),
    yaxis=list(title="Y",zeroline=F, range=c(0, 11)))

legend.plot <- plot_ly() %>% 
    add_markers(x = 1, 
    y = seq_len(length(unique_vals)),
    size = sort(unique_vals),
    showlegend = F, 
    marker = list(sizeref=0.1,sizemode="area")) %>%
    layout(
        annotations = list(
            list(x = 0.2, 
                y = 1, 
                text = "LEGEND TITLE", 
                showarrow = F, 
                xref='paper', 
                yref='paper')),
         xaxis = list(
            zeroline=F,
            showline=F,
            showticklabels=F,
            showgrid=F),
         yaxis=list(
            showgrid=F,
            tickmode = "array",
            tickvals = seq_len(length(unique_vals)),
            ticktext = sort(unique_vals)))


subplot(legend.plot, main.plot, widths = c(0.1, 0.9), 
    titleX=TRUE, titleY=TRUE)

Answer 2

首先，您只是将唯一值传递给图例。如果您传入所有可能的值（即seq(min(x), max(x), by=1)，或者在本例中为seq_len(max(x))），则图例将显示整个范围。

其次，marker 参数中的sizeref 和sizemode 改变了计算点大小的方式。以下示例应生成更一致的图：

set.seed(1)
df <- data.frame(x=1:10,y=runif(10,1,10),group=c(rep("A",9),"B"),group.size=as.integer(runif(10,1,10)))

require(plotly)  
require(dplyr)  

a <- plot_ly(type='scatter',mode="markers",
    color=~df$group,
    x=~df$x,
    y=~df$y,
    size=df$group.size,
    marker = list(sizeref=0.1, sizemode="area"),
    data=df,
    showlegend=F) %>%
    layout(title="Title",
        xaxis=list(title="X",zeroline=F),
        yaxis=list(title="Y",zeroline=F))

b <- plot_ly() %>% add_markers(x = 1, y = seq_len(max(df$group.size)),
    size = seq_len(max(df$group.size)), 
    showlegend = F, 
    marker = list(sizeref=0.1, sizemode="area")) %>%
    layout(
        xaxis = list(zeroline=F,showline=F,showticklabels=F,showgrid=F),
        yaxis=list(showgrid=F))


subplot(b, a, widths = c(0.1, 0.9))