pandas习题 197：有多少个不重复SKU

看过来

《pandas 教程》持续更新中，提供建议、纠错、催更等加作者微信: gr99123（备注：pandas教程）和关注公众号「盖若」ID: gairuo。跟作者学习，请进入 Python学习课程。欢迎关注作者出版的书籍：《深入浅出Pandas》和《Python之光》。

（编码题）商品可在多仓库同时铺货，Excel 宽表每列是仓、值是SKU编码。

import pandas as pd
from io import StringIO

raw = """
仓A,仓B,仓C
SKU001,SKU001,SKU002
SKU002,SKU003,SKU004
SKU005,SKU001,SKU002
"""

df = pd.read_csv(StringIO(raw))
df
# ...

老板问：“到底有多少个不重复SKU？”需要得到以下结果：

'''
   仓库  SKU种数
0  仓A      3
1  仓B      2
2  仓C      2
'''

参考答案

代码如下：

(
    df.apply('nunique')
    .to_frame('SKU种数')
    .reset_index(names='仓库')
)

（完）

pandas习题 197：有多少个不重复SKU

参考答案

相关内容