How can I change the keys of a dictionary so that any prefix beginning with a certain string maps to the same value?

I have the following dictionary:

SEC_DICT = {
    'equitiesa': 'MC',
    'equitiesab': 'MC',
    'etfsa': 'ETF',
    'etfsab': 'ETF',
    'etfsabc': 'ETF',
    'maba': 'MA',
    'mabab': 'MA',
}  

I want to edit or create a new dictionary that that everything starting with equities for example maps to MC, so something like SEC_DICT['equitiesblahblahblah'] would map to MC. Similar with etf and mab in the above dictionary.

The one catch is that the SEC_DICT is referrenced in many many places, so I would ideally not want to create something separate, because this would mean changing things in all places which reference this dictionary.

Is this possible?

For example, if I have the following function:

classify_sec(): 
    a = 'equitieshelloworld' 
    b = 'equitiesblahblahblah'
    y = SEC_DICT[a] 
    z = SEC_DICT[b] 

    return y, z 

I would expect the above to return MC, MC.

May a dictionary is NOT the right data_structure, because I don't want to list out all of the possibilities as keys, because in fact I don't know what the input is, I just want a generic structure where the mapping is something like: 'equities....' -> 'MC' for example.

Why not just have a function? Something like:

def SEC_DICT(key: str) -> str:
    values = {
        'equities': 'MC',
        'etf': 'ETF',
        'mab': 'MA',
    }
    for k in values:
        if key.startswith(k):
            return values[k]
    return ""

In the values dictionary, you provide what the "key" has to start with and what it should return. If the value isn't found, it returns an empty string.

Here is a solution that only requires updating the definition of the dictionary with a class you can include - any place that uses the dict will now work with partial keys (longest matching prefix):

from warnings import warn


class KeyPrefixDict(dict):
    def __getitem__(self, item):
        if not isinstance(item, str):
            warn(f"Key {item} is not a string")
            return super().__getitem__(item)
        else:
            # find the key that shares the longest prefix with item
            b_n, b_key = -1, ''
            for key in self.keys():
                n = 0
                for x, y in zip(key, item):
                    if x == y:
                        n += 1
                    else:
                        break
                if n > b_n:
                    b_n, b_key = n, key
                    # if the length of the prefix is the length if item, this is the best match
                    if n == len(item):
                        break
            # return the item at the best found key, or the item at item if no key with matching prefix was found
            return super().__getitem__(b_key if b_key else item)


SEC_DICT = KeyPrefixDict({
    'equitiesa': 'MC',
    'equitiesab': 'MC',
    'etfsa': 'ETF',
    'etfsab': 'ETF',
    'etfsabc': 'ETF',
    'maba': 'MA',
    'mabab': 'MA',
})

print(SEC_DICT['equities'])
print(SEC_DICT['mababcdefg'])
print(SEC_DICT['etfxyz'])
print(SEC_DICT['xyz'])  # longest shared prefix is '', so first item is matched
print(SEC_DICT[''])  # same

Output:

MC
MA
ETF
MC
MC

If you don't like the last two results and would prefer for a KeyError to be thrown when there is no matching prefix at all, change this line:

b_n, b_key = -1, ''

To:

b_n, b_key = 0, item

You can of course opt to leave the warn out altogether.

Use a for loop to iterate through the dictionary, something like this:

SEC_DICT = {
    'equitiesa': 'MC',
    'equitiesab': 'MC',
    'etfsa': 'ETF',
    'etfsab': 'ETF',
    'etfsabc': 'ETF',
    'maba': 'MA',
    'mabab': 'MA',
}

for key in SEC_DICT.keys():
    if key.startswith('equities'):
        SEC_DICT[key] = 'equ'
    elif key.startswith('etfs'):
        SEC_DICT[key] = 'etf'
    elif key.startswith('mab'):
        SEC_DICT[key] = 'ma'

print(SEC_DICT)

This code iterates through the dictionary, and checks the following conditions:

• if the key starts with equities, it changes it to equ.
• if the key starts with etfs, it changes it to etf.
• if the key starts with mab, it changes it to ma.