如何使用python/pandas获取透视表中一行的索引号?

1sbrub3j  于 2023-04-19  发布在  Python
关注(0)|答案(1)|浏览(189)

所以我有这个数据透视表,'movieUser_df':

而且我需要能够根据userID获取任何行的索引号。
我可以很容易地找到该行,如下所示:

movieUser_df.loc[movieUser_df.index.values == "641c87d06a97e629837fc079"]

但它只返回行数据。
我以为只需要movieUser_df.index == "641c87d06a97e629837fc079"就可以了,但它只返回了这个:

我假设这是一个布尔值,用于判断该行的每一列是否为0,但由于某种原因,它是相反的(因为返回数组中的第一项为true,尽管movieUser_df中唯一一个具有非0值的列是该行的最后一个数据点),但我不确定。
我四处寻找解决方案/其他人解决这个问题,但我能找到的只是this,但除非我错过了什么,否则他们最终会在问题主体中问一个完全不同的问题。
无论如何,有谁知道在这种情况下如何通过userID获得行索引号吗?如果有任何更多的信息可以提供帮助,请告诉我,谢谢您的帮助。

e5nqia27

e5nqia271#

Related post。在你的 Dataframe 中,userID是索引。要获得行数,你可以use numpy

df = pd.DataFrame([[1, 2], [4, 5], [7, 8]],
  index=['user1', 'user2', 'user3'],
  columns=['tt1', 'tt2'])

np.argmax(df.index == "user1") 
# 0

相关问题