gpt4 book ai didi

python - 如果找不到特定字符串,如何遍历数据框列表并删除所有数据

转载 作者:太空宇宙 更新时间:2023-11-04 02:04:17 25 4
gpt4 key购买 nike

我正在使用 python 库 Camelot 来解析多个 PDF 并提取这些 PDF 文件中的所有表格。第一行代码以列表格式返回从 pdf 中抓取的所有表格。我正在寻找一个特别的表,其中有一个唯一的字符串。值得庆幸的是,这个字符串对于这个表是唯一的,所以理论上我可以用它来隔离我想要抓取的表。

这些 pdf 或多或少是以相同的格式创建的,但是存在足够多的差异,我不能只对我想要的表进行静态调用。比如有的时候我要的表会是第一个爬出来的表,有的时候会是第三个。因此,我需要编写一些代码才能动态选择表格。

我脑海中的工作流程在逻辑上是这样的:

在 for 循环之前创建一个空列表以将表附加到。调用 for 循环并遍历 Camelot 代码输出的列表中的每个表。如果表中没有我要查找的字符串,则删除该表中的所有数据,然后将空数据框追加到空列表中。如果它确实有我要查找的字符串,则将其附加到空列表而不删除任何内容。

有没有更好的方法来解决这个问题?我确定可能有。

我已经将到目前为止的内容放在了我的代码中。如果字符串存在,我很难将条件语句放在一起以删除数据帧的所有行。如果字符串存在,我发现了很多删除列和行的示例,但没有删除整个数据框

import camelot
import pandas as pd

#this creates a list of all the tables that Camelot scrapes from the pdf
tables = camelot.read_pdf('pdffile', flavor ='stream', pages = '1-end')

#empty list to append the tables to
elist = []

for t in tables:
dftemp = t.df

#my attempt at dropping all the value if the unique value isnt found. THIS DOESNT WORK
dftemp[dftemp.values != "Unique Value", dftemp.iloc[0:0]]

#append to the list
elist.append(dftemp)

#combine all the dataframes in the list into one dataframe
dfcombined = pd.concat(elist)

最佳答案

您可以在 dftemp.values 返回的 numpy 数组上使用“in”运算符 link

for t in tables:
dftemp = t.df

#my attempt
if "Unique Value" in dftemp.values:
#append to the list
elist.append(dftemp)

关于python - 如果找不到特定字符串,如何遍历数据框列表并删除所有数据,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/55052989/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com