python - aggfunc 获取 pandas 数据透视表单元格中的任意值

Question

我想使用pivot_table来显示每个单元格中列的任意值。例如，给定一个像这样的 DataFrame:

df = pd.DataFrame({'x': ['x1', 'x1', 'x2'],
                   'y': ['a', 'b', 'c']})

要计算每个 x 值的 y 值:

df.pivot_table(index='x', values='y', aggfunc=len)
    y
x   
x1  2
x2  1

因此，我想用 ['a', 'c'] 或 ['b' 代替 [2, 1] ，'c']。

我尝试了这些方法，但都产生错误( notebook ):

df.pivot_table(index='x', values='y', aggfunc=sample)
df.pivot_table(index='x', values='y', aggfunc=head)
df.pivot_table(index='x', values='y', aggfunc=lambda x: x[0])

每https://stackoverflow.com/a/38982172/1840471 ，另一种方法是使用 groupby 和 agg，在这种情况下会产生所需的结果:

df.groupby(['x']).y.agg('head')

但是，我希望使用pivot_table，因为我的完整用例涉及获取行和列中的值。

Answer 1

如何使用first，如下所示:

df.pivot_table(index='x', values='y', aggfunc='first')

Out[67]:
    y
x
x1  a
x2  c