r语言 - 堆叠条形图，为单个单元格着色

Question

我正在尝试创建一个堆叠的条形图，但单个单元格根据单独的值着色。这是我正在使用的示例数据集：

count sample label
50      A    DD
20      A    BB
20      A    CC
10      A    AA
20      B    JJ
20      B    LL
10      B    BB
15      B    CC
30      B    AA
5       B    HH

我使用以下代码来创建绘图并为我要着色的单元格着色，但它为错误的单元格着色。我不确定为什么：

Library(lattice)
data$rank <- rank(data$count)
colVec <- rep("white", nrow(data))
colVec[data$label == "AA"] <- "red"
barchart(count ~ sample, groups = as.factor(rank), stack = TRUE, data = data, 
    as.table = TRUE, col = colVec, ylab = "Counts")

在此示例中，我希望样本 B 中的顶部栏和样本 A 中的最后一个条形图为红色（带有标签 AA 的单元格），但它最终为 5 个完全不同的单元格着色。

我希望使用此代码的真实数据集有数千个示例条目，因此我无法使用手动方法。

Answer 1

您希望每个数据点都有一个"组"，按计数排序。您只需要将颜色矢量重新排序为相同的顺序，在此处完成索引即可。

g <- order(data$count)
colVec <- ifelse(data$label=="AA","red","white")[g]
barchart(count ~ sample, groups = factor(1:10,levels=g), 
  stack = TRUE, data = data, col=colVec)

Answer 2

组部分定义条形图的堆叠部分的命名和排序方式。 因此，您需要创建一个新的分组向量，该向量为每个框具有唯一的名称，并按所需的顺序进行排序。 然后，您可以适当地设置颜色矢量。

data$label2 <- paste(data$label, data$sample)
data$label2 <- reorder(data$label2, data$rank)
colVec = rep("white", length(levels(data$label2)))
colVec[grepl('AA', levels(data$label2))] = "red"
barchart(count ~ sample, groups = label2, stack = TRUE, data = data, as.table = TRUE, ylab = "Counts", col=colVec)