python - pandas pivot_table 中的意外结果

Question

我正在尝试在 pandas Dataframe 上创建数据透视表。我几乎得到了预期的结果，但它似乎乘以了二。我可以除以二然后收工，但是，我想知道我是否做错了什么。

代码如下:

import pandas as pd
import numpy as np

df = pd.DataFrame(data={"IND":[1,2,3,4,5,1,5,5],"DATA":[2,3,4,2,10,4,3,3]})
df_pvt = pd.pivot_table(df, aggfunc=np.size, index=["IND"], columns="DATA")

df_pvt 现在是:

DATA   2    3    4    10
IND                     
1     2.0  NaN  2.0  NaN
2     NaN  2.0  NaN  NaN
3     NaN  NaN  2.0  NaN
4     2.0  NaN  NaN  NaN
5     NaN  4.0  NaN  2.0

但是，应该是 1.0 而不是 2.0!我误解了什么/做错了什么？

Answer 1

改用字符串 'size'。这将触发 Pandas 对“大小”的解释，即组中元素的数量。 NumPy 对大小的解释是每个维度长度的乘积。

df = pd.pivot_table(df, aggfunc='size', index=["IND"], columns="DATA")

print(df)

DATA   2    3    4    10
IND                     
1     1.0  NaN  1.0  NaN
2     NaN  1.0  NaN  NaN
3     NaN  NaN  1.0  NaN
4     1.0  NaN  NaN  NaN
5     NaN  2.0  NaN  1.0