从列表中划分字母数字词并将其存储为字典的键值对

Question

I have a list of alphanumeric data,

my_list = ["A1B2244", "B3H7654", "A1O6541", "J4777"]

I need to divide each word in dict form like

{"A1": ["B2244", "O6541"], "B3": ["H7654"], "J4": ["777"]}

Could you please let me know the easiest way to do this in python.

Answer 1

my_list = ['A1B2244', 'B3H7654', 'A1O6541', 'J4777']
my_dict={i[:2]:i[2:] for i in my_list}

Edit: Sorry I didn't notice the replication in your output. Others have short solutions, but a pure pythonic way is:

my_list = ['A1B2244', 'B3H7654', 'A1O6541', 'J4777']
my_dict={}
for i in my_list:
    if i[:2] in my_dict:
        my_dict[i[:2]].append(i[2:])
    else:
        my_dict[i[:2]]=[i[2:]]

Answer 2

You can use itertools.groupby to group the elements of your list based on your condition(first two characters). Then supply the result to dict constructor

>>> from itertools import groupby
>>> dict([(k,list(g)) for k,g in groupby(sorted(k),key=lambda x: x[:2])])
>>> {'J4': ['J4777'], 'A1': ['A1B2244', 'A1O6541'], 'B3': ['B3H7654']}

Answer 3

list = ['A1B2244', 'B3H7654', 'A1O6541', 'J4777']
#first initialize lists based 2 first elements
d= {i[:2]:[] for i in list}
#loop to add items by key
[d.get(i[:2]).append(i[2:]) for i in list]
print(d)

output:

{'A1': ['B2244', 'O6541'], 'J4': ['777'], 'B3': ['H7654']}

Answer 4

Just to add to the first answer by Willian Vieira, I thought it would be helpful to know the output print(d) right after d= {i[:2]:[] for i in list} , which is:

{'A1': [], 'B3': [], 'J4': []}

just to clarify this line in the two-line solution. Just to see how the keys are made from taking the first two characters in each element of the list (without duplication of these characters), and the values are initialized as empty lists.

Answer 5

As per your comment to the question, rule to split is: split after the first digit. So, you can search for the index of the first digit, split and add to the dict. I ignored an input with no digits.

def first_index_of_digit(st):
    for i in range(len(st)):
        if st[i].isdigit():
            return i
    return -1

my_list = ["A1B2244", "B3H7654", "A1O6541", "J4777"]

dd = dict()

for item in my_list:
    i = first_index_of_digit(item)
    if (i == -1):
        continue
    k, v = item[:i+1], item[i+1:]

    if (dd.get(k, 0) == 0):
        dd[k] = list()
    dd[k].append(v)

print(dd)
# {'A1': ['B2244', 'O6541'], 'B3': ['H7654'], 'J4': ['777']}