python - 如何将列表应用于 Pandas 组

<分区>

我看到了这个问题，grouping rows in list in pandas groupby

但我有两个以上的列要应用 list。

输入:

df(pd.DataFrame)

| index | c1 | c2 | c3 |
|-------|----|----|----|
|     1 | A  |  6 |  1 |
|     2 | A  |  5 |  2 |
|     3 | B  |  4 |  3 |
|     4 | B  |  3 |  4 |
|     5 | B  |  2 |  5 |
|     6 | C  |  1 |  6 |

预期输出:

| c1 |    c2   |    c3   |
|----|---------|---------|
| A  | [6,5]   | [1,2]   |
| B  | [4,3,2] | [3,4,5] |
| C  | [1]     | [6]     |

我也试过了

df.groupby('c1').apply(list)

但结果如下。

| c1 |             |
|----|-------------|
| A  | ['c2','c3'] |
| B  | ['c2','c3'] |
| C  | ['c2','c3'] |

我该怎么做？

谢谢。

最佳答案

众所周知 apply 与 list 的问题

df.groupby('c1').agg(lambda x : list(x))
Out[15]: 
           c2         c3
c1                      
A      [6, 5]     [1, 2]
B   [4, 3, 2]  [3, 4, 5]
C         [1]        [6]

关于python - 如何将列表应用于 Pandas 组，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/50653244/

上一篇：python - Pandas to_datetime() 没有检测到列

下一篇：python - Python 中列表的状态信息和不变性

相关文章：

python - 弃用 DataFrame.ix 意味着什么？

python - Pandas to_sql 设置列类型

python - 使用 Paramiko 在远程机器上执行命令无效

python - 如何覆盖 Gunicorn 的日志记录配置以使用自定义格式化程序

python:计算自然数的所有因素的列表

python-3.x - 如何获取对当前图形和轴对象的引用，以便我可以强制条形图 y 轴仅显示整数

python - 将数据帧转换为 numpy 数组时如何保持数据类型？

python - 使用 App Engine django 补丁提供简单图像？

python - 如何从 Tkinter 的弹出窗口中获取值？

python - 类型错误 : 'KeysView' object does not support indexing