我有一个关于一些艺术家,他们的专辑和他们的曲目的一些信息的数据框架。
df = pd.DataFrame({'Artist': ['A', 'A', 'A', 'B', 'B', 'C', 'C', 'C', 'C', 'D', 'E'], 'AlbumId': [201, 201, 451, 390, 390, 272, 272, 698, 698, 235, 312], 'TrackId': [1022, 3472, 9866, 6078, 2634, 3411, 8673, 2543, 5837, 9874, 1089]})
艺术家A有2张专辑(201和451),其中一张专辑有2首曲目(1022和3472),另一张专辑有1首曲目(9866)。
艺术家B有1张专辑(390)和2首曲目(6078和2634)。
艺术家C有2张专辑(272和698),每张专辑有2首曲目。
艺术家D有1张专辑(235)和1首曲目(9874)。
艺术家E有1张专辑(312)和1首曲目(1089)。
我想找到拥有超过1张专辑的艺术家,并相应地获得这些艺术家的行。我想要的输出如下所示:
我试过:
groupedArtists = data.groupby(['ArtistId', 'AlbumId']).filter(lambda group: (group.AlbumId.nunique() > 1))
但它似乎并不像预期的那样工作。
有人能帮帮我吗?我很感激!
5条答案
按热度按时间omhiaaxx1#
您希望仅按
ArtistId
而不是AlbumId
分组:输出:
x33g5p2x2#
分组应单独由 * 艺术家 * 进行。
然后,对于每个组,检查它包含多少(不同的)相册,并仅选取具有1个以上相册的组。
所以正确的解决方案是:
fnx2tebb3#
这是我找到的解决方案,它有点冗长,但可能更容易理解:
h9vpoimq4#
一个迟来的响应,但其他解决方案的替代方案如下:
与其他方法一样,该方法按艺术家分组,但不应用过滤函数,而是为每个组返回
pd.Series
数量的唯一专辑,并仅返回nunique
不等于1的那些行。在运行时间的比较中,这稍微快一点:
jgwigjjp5#
您可以根据相册的计数创建聚合 Dataframe ,然后根据所需相册的数量进行筛选