python - 将 Dataframes 与 .eq() 进行比较后获取索引

Question

对于 Web 抓取应用程序，我将我的数据库中的一些数据与我从网站上抓取的一些数据进行比较。我将数据保存在 Dataframe 的两个不同列中。

比较效果很好!我得到在 2 列中具有相同值的行数。但是随着我的项目不断增长，我也想知道比较在哪个索引上返回 true。我该怎么做？

一些附加信息

我的数据框:

df_single["Database"]:
Schloss Haindorf Hotelbetriebs GmbH
1. Aichfelder Druck Gesellschaft m.b.H.
Rössler Elektro Korbel
Elektro Schefbänker 
AWESOME X e.U.


df_single["Scraped"]:
Schloss Haindorf Hotelbetriebs GmbH
1. Aichfelder Druck Gesellschaft m.b.H.
Elektro Rössler OG
Elektro Schefbänker KG
AWESOME X e.U. 

我与.eq()的比较

same_single = df_single["Database"].str.lower().eq(df_single["Scraped"].str.lower()).sum()

我的输出:

[IN:] print(same_single)
[OUT:] 3

想要的输出:

[IN:] print(index where comparison = true)
[OUT:]Comparison was true at Index: 3,5 and 7

Answer 1

首先通过掩码过滤索引值到idx，然后将值与分隔符连接在一起:

mask = df_single["Database"].str.lower().eq(df_single["Scraped"].str.lower())

idx = df_single.index[mask]

print (f"Comparison was true at Index: {', '.join(idx.astype(str))}")

或者:

print ("Comparison was true at Index: {}".format(', '.join(idx.astype(str))))