控制聚合结果的dtype

Question

当我这样做时

df.groupby('id').aggregate({
    "timestamp": {
        "len" : len,
        ...
    },
    ....
})

我得到 datetime64 类型的 timestamp.len 列，显然，不是我想要的。

我该如何控制？

我大概可以做一些后期处理，比如

res[('timestamp','len')].astype(int)

但我宁愿马上得到正确的类型。

Answer 1

是的！这很奇怪。
请改用size

df = pd.DataFrame(dict(id=['a', 'a', 'b', 'b'],
                       timestamp=pd.date_range('2016-09-29', periods=4)))

df.groupby('id').aggregate({'timestamp': {'len': 'size'}})