gpt4 book ai didi

python - 按值获取列表中具有重复列名称的列表

转载 作者:行者123 更新时间:2023-12-04 00:01:10 25 4
gpt4 key购买 nike

我有数据框:

import pandas as pd

data = [[101, 1, 2, 10, 3, 2, 3, 1], [5,5, 5, 5, 5, 5, 5, 5], [30, 3, 7, 14, 10, 7, 10, 2], [11, 2, 6, 15, 20, 6, 20, 11]]

df = pd.DataFrame(data, columns = ['xen', 'sim', 'tab', 'sim', 'simm', 'box', 'simm', 'res'])

看起来像:

|   xen |   sim |   tab |   sim |   simm |   box |   simm |   res |
|------:|------:|------:|------:|-------:|------:|-------:|------:|
| 101 | 1 | 2 | 10 | 3 | 2 | 3 | 1 |
| 5 | 5 | 5 | 5 | 5 | 5 | 5 | 5 |
| 30 | 3 | 7 | 14 | 10 | 7 | 10 | 2 |
| 11 | 2 | 6 | 15 | 20 | 6 | 20 | 11 |

我需要在列表中按值获取具有重复列名称的列表(它可以按名称重复,也可以不重复)。对于上面输出的数据框应该是这样的:

[["tab", "box"], ["simm", "simm"]]

最佳答案

IIUC

l=df.T.reset_index().groupby(df.index.tolist())['index'].agg(list).loc[lambda x : x.str.len()>=2].values.tolist()
[['tab', 'box'], ['simm', 'simm']]

关于python - 按值获取列表中具有重复列名称的列表,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/61260149/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com