寻找一种在python numpy中堆叠图像以生成数据集的方法

Question

我正在尝试堆叠numpy数组（图像）。我每个都有大约1000张图像（20,20），我想将所有图像堆叠到一个阵列中。目前，我正在使用np.dstack。它给我一个像（20,20,1000）的数组。我希望结果喜欢是（1000,20,20）。

Answer 1

通过使用numpy.stack，可以指定阵列堆叠的轴。

文档中的示例：

mydat %>% 
  group_by(ItemRelation, DocumentNum) %>% 
  summarise(X1 = sum(SaleCount[SaleCount > 0 & IsPromo == 0]), 
            X2 = sum(SaleCount[SaleCount < 0 & IsPromo == 0]), 
            X3 = X1 + X2, 
            X4 = ifelse(sum(SaleCount > 0 & IsPromo == 0)==0, 0, X3/sum(SaleCount > 0 & IsPromo == 0)),
            X5 = sum(SaleCount[IsPromo == 1]),
            X6 = ifelse(sum(SaleCount > 0 & IsPromo == 1)==0, -X4,  
                        X5/sum(SaleCount > 0 & IsPromo == 1) - 
                        X3/sum(SaleCount > 0 & IsPromo == 0)*
                        sum(SaleCount > 0 & IsPromo == 1))) %>% 
  ungroup() %>%
  mutate_all(~ifelse(. %in% c(-Inf, Inf), 0, .))