在pandas中,可以使用groupby()
函数将DataFrame中的列值进行分组,并将其他列值放入列表中。具体步骤如下:
import pandas as pd
df
的DataFrame,包含两列col1
和col2
。groupby()
函数进行分组:grouped = df.groupby('col1')
apply()
函数结合lambda
表达式来实现。例如,将col2
的值放入列表中:grouped.apply(lambda x: x['col2'].tolist())
完整代码示例:
import pandas as pd
# 创建DataFrame
data = {'col1': ['A', 'A', 'B', 'B', 'A'],
'col2': [1, 2, 3, 4, 5]}
df = pd.DataFrame(data)
# 使用groupby()函数进行分组
grouped = df.groupby('col1')
# 遍历分组并将其他列值放入列表中
result = grouped.apply(lambda x: x['col2'].tolist())
print(result)
输出结果为:
col1
A [1, 2, 5]
B [3, 4]
dtype: object
在这个例子中,我们根据col1
列的值进行分组,然后将每个分组中的col2
列的值放入列表中。最后,输出了每个分组对应的列表。
腾讯云相关产品和产品介绍链接地址:
领取专属 10元无门槛券
手把手带您无忧上云