python - 获取列与某个值匹配的行的索引

Question

论坛里有一个很相似的帖子 POST但我只是不知道在我的例子中该怎么做。

现在我的代码，解释如下

for i in dfuser.appid :
  print i
  d = dfbinary.loc[dfbinary['appid'] == i]
  print d
  glist = dfbinary.columns[dfbinary.loc[i]==1] 
  print glist

我有一个数据框，其中包含用户及其应用程序的列表(dfuser)，并且我有另一个数据框，其中包含所有应用程序的类型(一个应用程序可能有多个类型)。所以我想看看哪种类型在每个用户中更受欢迎。

我的代码很好，除了glist没有找到我想要的appid，而是找到索引为i的appid。例如 i=10 ，因此它将找到位于第 11(10) 行的应用程序。

这就是它打印的内容

   10
   appid  Accounting  Action  Adventure  Animation&Modeling  AudioProduction... 
0   10.0         0.0     1.0        0.0                 0.0              0.0  
[1 rows x 23 columns]
Index([u'Action'], dtype='object') 

(这恰好是正确的)

Answer 1

首先，每当你与 pandas 进行循环时，你可能都做错了!

您需要使用合并来组合两个数据框，并仅选择用户和流派列。它的工作原理就像 SQL 一样。然后你就有了一个以用户/类型为键的表。现在您可以 groupby("user").count()。没有显式循环。