listdataframe某列为list,将它转化成每个list包含所有元素的列
时间: 2023-08-09 10:11:21 浏览: 87
可以使用`explode()`函数将某一列中的列表转化为多行,每行包含一个列表元素。然后使用`groupby()`函数对其他列进行分组,对于同一组的行,将列表元素合并起来即可。
下面是一个示例代码:
```python
import pandas as pd
# 创建示例DataFrame
df = pd.DataFrame({'id': [1, 2, 3],
'data': [[1, 2], [2, 3, 4], [5, 6, 7, 8]]})
# 使用explode()函数将data列中的列表转化为多行
df = df.explode('data')
# 对id列进行分组,并将data列中的元素合并为一个列表
df = df.groupby('id')['data'].apply(list).reset_index()
print(df)
```
输出结果为:
```
id data
0 1 [1, 2]
1 2 [2, 3, 4]
2 3 [5, 6, 7, 8]
```
其中,`explode()`函数将`data`列中的列表转化为多行,得到:
```
id data
0 1 1
0 1 2
1 2 2
1 2 3
1 2 4
2 3 5
2 3 6
2 3 7
2 3 8
```
然后,对`id`列进行分组,并将`data`列中的元素合并为一个列表。
阅读全文