python表中的keyerror

Question

我对以下代码感到困惑：

class Table(object):
    def __init__(self):
        self.d = collections.defaultdict(dict)
    def add(self, row, col, val):
        self.d[row][col] = val
    def get(self, row, col, default=None):
        return self.d[row].get(col, default)
    def incol(self, col, val):
        for x,y in self.d.iteritems():
            if y[col]==val:
                wholerow=y
        return wholerow.values()

创建表后，无法使用“ incol”函数来获取所有值。 它总是给我关键错误。 但是，当我只使用相同的代码而不使用函数时，我可以得到想要的东西。 我的代码发生了什么...。

Answer 1

上面的代码并不“总是给我键错误”（“总是”是我在这里争论的词）。 例如，考虑以下代码：

foo = Table()
foo.add("row_1", "col_1", "col_1_value")
foo.add("row_1", "col_2", "col_2_value")
print foo.get("row_1", "col_1")
print foo.incol("col_1", "col_1_value")

这可以正常工作，并产生以下输出：

col_1_value
['col_2_value', 'col_1_value']

但是，如果你试图通过一个不存在的列名incol ，你会得到一个KeyError 。 例如，如果将以下行添加到上面的代码中：

print foo.incol("i do not exist", "my imaginary value")

您最终将收到类似于以下内容的错误消息：

Traceback (most recent call last):
File "./stack_overflow-2013-03-14.py", line 21, in <module>
    print foo.incol("i do not exist", "my imaginary value")
File "./stack_overflow-2013-03-14.py", line 12, in incol
    if y[col]==val:
KeyError: 'i do not exist'

可以处理这种情况的incol示例实现为：

def incol(self, col, val):
    result = None
    for x,y in self.d.iteritems():
        if col in y and y[col]==val:
            result = y.values()
            break
    return result

现在，如果您运行以下行：

foo = Table()
foo.add("row_1", "col_1", "col_1_value")
foo.add("row_1", "col_2", "col_2_value")
print foo.get("row_1", "col_1")
print foo.incol("col_1", "col_1_value")
print foo.incol("i do not exist", "my imaginary value")
print foo.incol("i do not exist", None)

你得到：

col_1_value
['col_2_value', 'col_1_value']
None
None

您还可以使用Martijn Pieters建议的以下incol实现：

def incol(self, col, val):
    result = None
    for x,y in self.d.iteritems():
        if y.get(col)==val:
            result = y.values()
            break
    return result

但是在这种情况下，如果您传递不存在的col名称并且val = None ，那么它将匹配任何行。 相应的输出将是：

col_1_value
['col_2_value', 'col_1_value']
None
['col_2_value', 'col_1_value']

也许这就是您想要的应用程序行为。 那是你决定的。