dictionary - 从字典中递增和分配 id 的 Pythonic 方法

Question

这似乎是一个很常见的模式:

for row in reader:
    c1=row[0]
    if ids.has_key(c1):
        id1=ids.get(c1)
    else:
        currid+=1
        id1=currid
        ids[c1]=currid

我想知道是否有更好的方法来实现这一点。就单行 if 语句而言，我可以做很多事情:

id1=ids.get(c1) if ids.has_key(c1) else currid+1

但是我坚持递增 currid 并在执行 else 情况时坚持并在 if 条件通过时将 c->id1 坚持到字典中。

Answer 1

如果ids从0开始:

for row in reader:
    id1 = ids.setdefault(row[0], len(ids))

(另外:has_key 被认为已弃用。使用 x in d 而不是 d.has_key(x)。)