python - Pandas df.idmax 在 0.22.0 版本中失败

Question

Pandas 版本:0.22.0
我的代码在 0.20 版本中工作

    pvt = df.pivot_table(columns='period', values='qty', aggfunc='sum')
    df['qty'] = df['qty'].astype(float)
    print pvt
    d = pvt.idxmax(axis = 1)

pvt.idmax 在我之前的版本中运行良好。
现在它给出错误“此数据类型不允许减少操作'argmax'”
我需要降级我的 pandas 版本吗？

Answer 1

您应该在旋转 df 之前应用 .astype(float)。也许新版本的 pandas 对 dtypes 更加严格。试试这个:

df['qty'] = df['qty'].astype(float)
pvt = df.pivot_table(columns='period', values='qty', aggfunc='sum')
print pvt
d = pvt.idxmax(axis = 1)