How to match two arrays

前端 未结 3 851
天命终不由人
天命终不由人 2021-01-26 11:51

I have two arrays

A = [a, b, c, d]

and

B = [a1, a2, b1, b2, b3, c1, c2, c3, d1, d2, d3, d4]

I want to match b

3条回答
  •  忘掉有多难
    2021-01-26 12:02

    from collections import defaultdict
    
    A = ["a", "b", "c", "d"]
    B = ["a1", "a2", "b1", "b2", "b3", "c1", "c2", "c3", "d1", "d2", "d3", "d4"]
    d = defaultdict(list)
    for item in B:
        prefix = item[0]
        if prefix in A:
            d[prefix].append(item)
    

提交回复
热议问题