3

我刚刚阅读了一堆关于如何在 Python 中处理 StopIteration 错误的帖子，但是我在解决我的特定示例时遇到了麻烦。基本上，我有一个带有很多前缀的 csv 文件。此文件有两列带标题：字数和计数。计数是该前缀出现的频率。我还有另一个包含公司名称列表的文件。前缀文件从公司文件中每个公司名称的第一个单词中获取前缀。我正在尝试删除重复项，而我现在想要做的是：

每次发生此错误时忽略 StopIteration 错误。

换句话说，我不必在下面写下所有注释掉的“if”语句，我只想要一行说：如果生成了 StopIteration 错误，只需通过将有问题的“前缀”视为某种方式来忽略该错误，就好像它是前缀文件中出现两次以上的前缀，因此我们应该返回不包含前缀的公司名称的值。我意识到这忽略了前缀文件中存在不同的前缀值和公司名称的实际前缀这一事实，但通常它与 python 和 excel 之间存储的非美式英文字母以及其他一些看起来不是特别系统的方式，所以我稍后会手动删除它们。

我的代码是：

def remove_prefix(prefix, first_name):
   #try:
   #EXCEPTIONS:
   #if '(' in prefix:
   #    prefix = prefix[1:]
      #if ')' in prefix:
    #  prefix = prefix[:-1]
    """
      if prefix == "2-10":
        prefix = "2"
      if prefix == "4:2:2":
        prefix = "4"
      if prefix == "5/0" or prefix == "5/7" or prefix == "58921-":
        prefix = "5"
    """
    #except StopIteration:
   #    pass

   print(first_name, prefix)
   input_fields = ('Word', 'Count')
   reader = csv.DictReader(infile1, fieldnames = input_fields)
   #if the prefix has a frequency of x >=2 in the prefix file, then return first_name  without prefix
   #else, return first_Name
   infile1.seek(0)
   #print(infile1.seek(0))
   next(reader)
   first_row = next(reader)
   while prefix != first_row['Word'] and prefix[1:]!= first_row['Word']:
      first_row = next(reader)
      #print(first_name, prefix)
      #print(first_row, first_name, prefix, '\t' + first_row['Word'], prefix[1:])
   if first_row['Count'] >= 2:
      length = len(prefix)
      first_name = first_name[length+1:]
  #print("first name is ", first_name)
  return first_name
4

2 回答 2

3

我不认为这是由你认为的原因造成的。StopIteration 异常是在生成器 ( reader) 用完要读取的行时引起的。

例如：

def g():
    "generates 1 (once)"
    yield 1

a = g()
next(a) # is 1
next(a) # StopIteration exception (nothing left to yield)

要解决此问题，您可以next尝试将其包装起来，除了（通过）：

while prefix != first_row['Word'] and prefix[1:]!= first_row['Word']:
    try:
        first_row = next(reader)
    except StopIteration:
        pass

然而，正如大卫指出的那样，这可能不是你应该这样做的方式。

于 2012-08-31T19:47:07.627 回答
2

这可以通过首先从文件中创建前缀列表，然后startswith在每个前缀上使用该方法来以更简单的方式完成。例如：

reader = csv.DictReader(infile1)
# # this is assuming there are only two columns in the file: Word and Count
prefixes = [l["Word"] for l in list(reader) if int(l["Count"]) >= 2]

def remove_prefix(first_name):
    for p in prefixes:
        if first_name.startswith(p):
            return first_name[len(p):]
    return first_name

那不是更简单吗？另一个优点是它只读取文件一次，而不是为要替换的每个单词重新打开它。

于 2012-08-31T19:39:49.950 回答