通过子字符串分配Python字典

2019-10-25 10:58:14 阅读：169 来源： 互联网

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

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

如果仅提供密钥的子字符串或大小写不同的版本,那么在此结构中为现有密钥插入新值的最有效方法是什么？例如. “抢”或“ 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

解决方法:

首先,正如我在问题下的注释中所解释的那样,您可以做一些事情来优化现有代码：

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))查找子字符串的索引.接下来,我们在长度数组中计数以找到相应的键.

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

标签：dictionary,indexing,python
来源： https://codeday.me/bug/20191025/1928093.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9

通过子字符串分配Python字典