在数据框中搜索字符串,并获取其中存在的列名称

时间:2019-09-20 21:18:11

标签: python-3.x dataframe

与数据框和查找值有关的大多数问题都适合于搜索序列和对这些相关行进行索引/切片。

我有一个要搜索的重复字符串,并且从技术上来说,我可以使用两个for循环来找到它,但是可以肯定的是,这样做一开始就破坏了使用数据帧的目的,并且大大降低了运行时间。我想在示例字符串“ abc”的行中找到第一个实例,并添加一个新列,并在其中显示该列的名称。

示例代码:

col = df.columns
for index,row in df.iterrows():
    for colName in col:
        if pivot_df.loc[index,colName]=="abc":
            pivot_df.loc[index,'end'] = str(colName)
            break

有没有更清洁的方法来找到此解决方案?

0 个答案:

没有答案