有了Pandas 群比,我可以做这样的事情:
>>> df = pd.DataFrame(
... {
... "A": ["foo", "bar", "bar", "foo", "bar"],
... "B": ["one", "two", "three", "four", "five"],
... }
... )
>>> print(df)
A B
0 foo one
1 bar two
2 bar three
3 foo four
4 bar five
>>> print(df.groupby('A')['B'].unique())
A
bar [two, three, five]
foo [one, four]
Name: B, dtype: object
我想要的是生成索引列表而不是B列列表的输出:
A
bar [1, 2, 4]
foo [0, 3]
然而,groupby('A').指数unique()不起作用.什么语法可以为我提供我想要的输出?我非常乐意用其他方式来做这件事,而不是使用groupby,尽管我确实需要在我的实际应用程序中按两列进行分组.