按逻辑标准拆分子列表中的列表

Question

我有一个列表，其中每个元素可以是一个数字或两个不同字符串之一（我们称之为“a”和“b”）。我想将列表分成三个列表：数字列表、“a”列表和“b”列表。

我可以使用条件列表理解来做到这一点：

mixlist=[1,2,'a',3,'b','a']
numlist=[el for el in mixlist if type(el)==int]
alist=[el for el in mixlist if el=='a']
blist=[el for el in mixlist if el=='b']

但它会导致在 mixlist 上运行多次。可能我可以做得更好，但我不知道只有一次在 mixlist 上运行的捷径......

Answer 1

你可以直接做，像这样

numlist, alist, blist = [], [], []
for item in mixlist:
    if item == "a":
        alist.append(item)
    elif item == "b":
        blist.append(item)
    else:
        numlist.append(item)

或者你可以把它缩短一点，像这样

numlist, alist, blist = [], [], []
d = {"a": alist, "b": blist}
for item in mixlist:
    d.get(item, numlist).append(item)

dict.get方法会返回key对应的值。但是当没有找到key时，它会返回作为第二个参数传递的默认值。因此，在我们的例子中，如果键是a 或b，我们将得到对应的alist 或blist。如果两者都不是，那么我们将得到 numlist 并将 item 添加到它。

编辑：您可以像这样稍微改进直接的方式

mixlist = [1, 2, 'a', 3, 'b', 'a']

numlist, alist, blist = [], [], []
nappend, aappend, bappend = numlist.append, alist.append, blist.append

for item in mixlist:
    if item == "a":
        aappend(item)
    elif item == "b":
        bappend(item)
    else:
        nappend(item)