开发者社区 问答 正文

如何在pandas数据帧中将副本限制为5?

col1= ['A','B','A','C','A','B','A','C','A','C','A','A','A'] col2= [1,1,4,2,4,5,6,3,1,5,2,1,1]

df = pd.DataFrame({'col1':col1, 'col2':col2})

对于A,我们有[1,4,4,6,1,2,1,1],8个条目,但是我想在将数据帧转换为dict/list时将大小限制为5

展开
收起
游客6qcs5bpxssri2 2019-09-22 15:00:13 1479 分享 版权
1 条回答
写回答
取消 提交回答
  • 使用pandas.DataFrame.groupby的apply:

    df.groupby('col1')['col2'].apply(lambda x:list(x.head(5))).to_dict()

    输出:

    {'A': [1, 4, 4, 6, 1], 'B': [1, 5], 'C': [2, 3, 5]}

    2019-09-22 15:02:01
    赞同 展开评论
问答分类:
问答标签:
问答地址: