使用`reindex`时显示空行

Posted 2023-03-12

【中文标题】使用`reindex`时显示空行

【英文标题】：Showing empty rows when using `reindex`

【发布时间】：2016-12-25 21:13:50

【问题描述】：

我在使用 reindex() 计算我的 DataFrame 中的各种组合时遇到了一些困难。

下面的代码重现了我的问题：

a = [
    ['Brand A' if i==0 else 'Brand B' for i in np.random.randint(0,2,size=(100,))],
    ['Type 1' if i==0 else 'Type 2' for i in np.random.randint(0,2,size=(100,))],
    ['Red' if i==0 else 'Blue' for i in np.random.randint(0,2,size=(100,))]
]
b = pd.DataFrame(a, index=['Brand', 'Type', 'Color']).T
b.loc[(b.Brand=='Brand A')&(b.Type=='Type 1'), 'Color'] = 'Red'   # no Blue, Type 1, Brand A
b.loc[(b.Brand=='Brand B')&(b.Type=='Type 2'), 'Color'] = 'Blue'  # no Red, Type 2, Brand B

c = b.groupby(['Brand','Type','Color'])
c.size()\
 .reindex(['Blue','Red'], level=2, fill_value=0)

输出：

Brand    Type    Color
Brand A  Type 1  Red      17
         Type 2  Blue     17
                 Red      19 
Brand B  Type 1  Blue     13
                 Red       9
         Type 2  Blue     25
dtype: int64

有没有办法得到这个输出：

Brand    Type    Color
Brand A  Type 1  Blue      0
                 Red      17
         Type 2  Blue     17
                 Red      19 
Brand B  Type 1  Blue     13
                 Red       9
         Type 2  Blue     25
                 Red       0
dtype: int64

【参考方案1】：

您可以使用unstack 和stack：

print (b.groupby(['Brand','Type','Color']).size().unstack(2, fill_value=0).stack())
Brand    Type    Color
Brand A  Type 1  Blue      0
                 Red      21
         Type 2  Blue     20
                 Red      14
Brand B  Type 1  Blue     15
                 Red      11
         Type 2  Blue     19
                 Red       0
dtype: int64

reindexMultiIndex.from_product 的解决方案：

iterables = [['Brand A', 'Brand B'], ['Type 1', 'Type 2'], ['Blue','Red']]
idx = pd.MultiIndex.from_product(iterables, names=['Brand', 'Type', 'Color'])
print (b.groupby(['Brand','Type','Color']).size().reindex(idx, fill_value=0))
Brand    Type    Color
Brand A  Type 1  Blue      0
                 Red      21
         Type 2  Blue     20
                 Red      14
Brand B  Type 1  Blue     15
                 Red      11
         Type 2  Blue     19
                 Red       0
dtype: int64

【讨论】：

我明白了，所以我的问题是我只重新索引一个级别，而不是直接用整个 MultiIndex 重新索引？感谢您的帮助！

是的，我尝试了所有可能的组合并且只为我工作这两种方法。