Python字典按子串赋值

Question

给定一个预先存在的 Python 字典:

my_dict = {
    "Robert" : 37,
    "Kevin"  : 25,
    "Mark"   : 81,
}

如果只为您提供子字符串或键的不同大小写版本，那么在此结构中为现有键插入新值的最有效方法是什么？例如。 “Rob”或“kev”

我目前所做的工作，但我正在做类似以下的事情，对于多个输入是 O(N^2):

for key in list(my_dict):
    if my_input.name.lower() in key.lower():
        my_dict[key] = my_input.value
        break

Answer 1

首先，正如我在问题下的评论中所解释的那样，您可以采取一些措施来优化现有代码:

my_dict = {
    "Robert" : 37,
    "Kevin"  : 25,
    "Mark"   : 81,
    "Andre"  : 55
}

# optimized version of existing code
def update_key(my_dict, my_input, my_value):
    my_input = my_input.lower()
    for key in my_dict:
        if my_input in key.lower():
            my_dict[key] = my_value
            break

作为@Stephen Rauch确实如此，我认为建立索引是可行的方法。但是，我认为构建每个键的每个可能的子字符串的成本很高。此外，它对出现在字符串中间或末尾的子字符串不稳健。

这是另一种基于索引的方法。这个想法是我们将所有键连接在一起并构建一个字符串。我们还跟踪列表中每个字符串的长度。

然后对于给定的测试字符串，我们使用str.find()(应该是O(N))找到子字符串的索引。接下来我们在长度数组中进行计数，找到对应的key。

def create_index(my_dict):
    key_str = ""
    key_lengths = []
    for key in my_dict:
        key_str += key
        key_lengths.append(len(key))

    # return the concatenated keys, a lower case version, and the lengths list
    return (key_str, key_str.lower(), key_lengths)

def key_lookup(key_str, key_str_lower, key_lengths, my_input):
    idx = key_str_lower.find(my_input.lower())
    if idx == -1:
        return None
    len_sum = 0
    for kl in key_lengths:
        if idx < len_sum+kl:
            return key_str[len_sum:len_sum+kl]
        else:
            len_sum += kl
    return None

key_str, key_str_lower, key_lengths = create_index(my_dict)

print key_str_lower
#andrerobertkevinmark
print key_lengths
#[5, 6, 5, 4]

示例输出:

print(key_lookup(key_str, key_str_lower, key_lengths, my_input='rob'))
#Robert
print(key_lookup(key_str, key_str_lower, key_lengths, my_input='dre'))
#Andre

注意事项

• 我认为您现有的代码可能比这个解决方案更好。它绝对更具可读性，也更容易理解。
• 也有一些方法可以修改它，具体取决于您的实际目的。例如，您可以将每个可能的偏移量以 {offset: key} 的形式存储在字典中，而不是遍历和计算字符串偏移量。这会稍微增加索引创建时间并使用更多空间，但查找速度会更快。
• 与原始示例一样，如果子字符串包含在多个字符串中，这将返回第一个匹配项。由于字典是无序的，因此在构建索引时可以选择对键进行排序，以便输出是确定的。

---

计时结果

import random
import string

random.seed(12345)

N = 1000
my_dict = {
    ''.join(random.choice(string.ascii_letters) for _ in range(random.randint(3,30))): j for j in range(N)
}

def key_lookup_orig(my_dict, my_input):
    my_input = my_input.lower()
    for key in my_dict:
        if my_input in key.lower():
            return key
    return None

subs = [k[random.randint(0,len(k)-1):max(random.randint(0,len(k)), len(k))] for k in my_dict]

%%timeit
sum([1 if not key_lookup_orig(my_dict, s) else 0 for s in subs])
#1000 loops, best of 3: 1.5 ms per loop

key_str, key_str_lower, key_lengths = create_index(my_dict)

%%timeit
sum([1 if not key_lookup(key_str, key_str_lower, key_lengths, my_input=s) else 0 for s in subs])
#1000 loops, best of 3: 795 µs per loop