如何從一個列表的Python

Question

我有一個叫做動物列表創建下劃線集列表，

animals = ["B_FOX", "A_CAT", "A_DOG", "A_MOUSE", 
     "B_DOG", "B_MOUSE", "C_DUCK", "C_FOX", "C_BIRD"] 

，並希望以下的輸出：

A = ["A_CAT", "A_DOG", "A_MOUSE"] 
B = ["B_DOG", "B_MOUSE", "B_FOX"] 
C = ["C_DUCK", "C_FOX", "C_BIRD"] 

我只能得到一個子集列表只有字母或像這樣的動物：

[species.split("_",1)[1] for species in animals] 
    ['FOX', 'CAT', 'DOG', 'MOUSE', 'DOG', 'MOUSE', 'DUCK', 'FOX', 'BIRD'] 

    [letters.split("_",1)[0] for letters in animals] 
    ['B', 'A', 'A', 'A', 'B', 'B', 'C', 'C', 'C'] 

不知道我是否正確表達了這個問題。任何幫助解決這個棘手的問題將不勝感激！

Answer 1

可以解開前綴的值和名稱都從一個調用split：

groups = {} 
for animal in animals: 
    prefix, name = animal.split("_") 
    if prefix not in groups: 
     groups[prefix] = [] 
    groups[prefix].append(animal) 

print groups 

{'A'：['A_CAT'，'A_DOG'，'A_MOUSE']，'C'：['C_DUCK'，'C_FOX'，'C_BIRD']，'B'：['B_FOX'，'B_DOG'， 'B_MOUSE']}

如果需要，您以後仍然可以解壓詞典到單個變量：

A = groups["A"] 
B = groups["B"] 
C = groups["C"] 

如果你想擺脫前綴：

groups = {} 
for animal in animals: 
    prefix, name = animal.split("_") 
    if prefix not in groups: 
     groups[prefix] = [] 
    groups[prefix].append(name) 

Answer 2

你可以建立單獨的列表，每個首字母一個，但是，如果你有很多字母，這將是棘手。您可以使用defaultdict代替：

from collections import defaultdict 

d = defaultdict(list) 
animals = ["B_FOX", "A_CAT", "A_DOG", "A_MOUSE", 
    "B_DOG", "B_MOUSE", "C_DUCK", "C_FOX", "C_BIRD"] 

for animal in animals: 
    d[animal[0]].append(animal) 
print(dict(d)) 

輸出：

{'A': ['A_CAT', 'A_DOG', 'A_MOUSE'], 'C': ['C_DUCK', 'C_FOX', 'C_BIRD'], 'B': ['B_FOX', 'B_DOG', 'B_MOUSE']} 

Answer 3

嘗試的itertools.groupby根據第一個字母：

import operator as op 
import itertools as it 


animals = [ 
    "B_FOX", "A_CAT", "A_DOG", "A_MOUSE", 
    "B_DOG", "B_MOUSE", "C_DUCK", "C_FOX", "C_BIRD" 
] 

A, B, C = [list(g) for _, g in it.groupby(sorted(animals), key=op.itemgetter(0))] 

輸出：

A 
# ['A_CAT', 'A_DOG', 'A_MOUSE'] 

B 
# ['B_DOG', 'B_FOX', 'B_MOUSE'] 

C 
# ['C_BIRD', 'C_DUCK', 'C_FOX'] 

這是關於groupby如何工作的post。