gpt4 book ai didi

python - 如何有效地从边列表创建邻接列表

转载 作者:行者123 更新时间:2023-11-30 22:49:54 26 4
gpt4 key购买 nike

我有一个看起来像这样的 csv

id1,id2
a,b
c,d
a,e
c,f
c,g

我使用 df = pd.read_csv("file.csv") 将其读入数据帧。

我想将其转换为邻接表。那就是输出应该是

a,b,e
c,d,f,g

我觉得 df.groupby('id1') 应该有帮助,但可变长度行不适合 pandas,所以我有点卡住了。然而,由于我的 csv 很大,我正在寻找有效的解决方案。

有什么好的方法可以做到这一点?

最佳答案

如果您需要 CSV 字符串:

In [107]: df.groupby('id1').id2.apply(lambda x: ','.join(x)).reset_index()
Out[107]:
id1 id2
0 a b,e
1 c d,f,g

关于python - 如何有效地从边列表创建邻接列表,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/39562253/

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