优化计数列表中元素之间的变化

Question

我有一些工作代码来跟踪列表元素之间的“变化”-这样，任何两个不相同的连续元素都会构成一个变化。 该代码在这里可能很容易理解。

testlist = ['red','red','blue','red','red','black','yellow','black','yellow','blue']

第一个red到red不会触发任何更改，但是接下来的red到blue将会触发。 我也想统计每种颜色的变化。

# Set Tally counters to 0 and a unique key
red = 0
blue = 0
black = 0
yellow = 0
key = 40006

for i in range(len(testlist)-1):
    if (testlist[i] == (testlist[i+1])):
        print("No Change")
    else:
        print("Change to: " + str(testlist[i+1]))
        if testlist[i+1] == 'red':
            red = red + 1
        elif testlist[i+1] == 'blue':
            blue = blue + 1
        elif testlist[i+1] == 'black':
            black = black + 1
        elif testlist[i+1] == 'yellow':
            yellow = yellow + 1
dictfordf = {'key':key, 'red':red,'blue':blue,'black':black,'yellow':yellow}

这可以正常工作并正确输出{'black': 2, 'blue': 2, 'key': 40006, 'red': 1, 'yellow': 2} 。

当唯一元素的数量增加（此示例中仅4种唯一颜色）到10时， if/elif变得非常冗长。

我的两个问题是：

• 有没有更简洁的方法来实现这一目标？
• 有没有更快的方法来执行此任务？

Answer 1

首先，由于您的目标是构建一个dict，所以只需动态构建dict，而不是构建一堆单独的变量，然后将它们放在dict的最后。

您也可以使用Counter而不是Plain dict，因此您不必担心检查颜色是否已经存在。

当我们使用它时，不需要在已经是字符串的东西上调用str ，并且到处都有很多不必要的括号。

所以：

from collections import Counter
dictfordf = Counter()
dictfordf['key'] = 40006
for i in range(len(testlist)-1):
    if testlist[i] == testlist[i+1]:
        print("No Change")
    else:
        print("Change to: " + testlist[i+1])
        dictfordf[testlist[i+1]] += 1

存储确实不是一个数字的'key'值有点麻烦，因此您可能需要考虑使用defaultdict或普通字典上的setdefault 。 但我认为这还不错。

当然，如果'key'可能是testlist中的元素之一，那么这将使key增加。 但是，如果可能的话，尚不清楚在这种情况下应该发生什么，因此也不清楚您要如何解决它。

---

同时，您可以通过遍历相邻对来简化操作。 请参阅itertools文档中的pairwise配方。 但这当然会在代码中添加pairwise定义（或者您可以从more-itertools或toolz类的第三方库中导入pairwise定义）。

所以：

from collections import Counter
from itertools import tee

def pairwise(iterable):
    "s -> (s0,s1), (s1,s2), (s2, s3), ..."
    a, b = tee(iterable)
    next(b, None)
    return zip(a, b)

dictfordf = Counter()
dictfordf['key'] = 40006
for prev, current in pairwise(testlist):
    if prev == current:
        print("No Change")
    else:
        print("Change to: " + current)
        dictfordf[current] += 1

---

您可以使用groupby或itertools的unique_justseen配方进一步抽象事物。 我认为这会模糊不清，而不是澄清在哪里print输出。但是，假设您了解pairwise版本，则有必要对它们进行阅读，并尝试编写两个替代方案，至少作为练习。

Answer 2

我对这个问题的看法：

from collections import Counter

testlist = ['red','red','blue','red','red','black','yellow','black','yellow','blue']

def changes(data):
    last = data[0]
    for i in data:
        if last != i:
            yield i
        last = i

c = Counter(changes(testlist))
c['key'] = 40006
print(dict(c))

输出：

{'yellow': 2, 'red': 1, 'key': 40006, 'blue': 2, 'black': 2}

Answer 3

我的拍摄使用zip来成对浏览列表，非常简洁。 像其他工具一样，它使用Counter ，我同意这是完成这项工作的正确工具。

from collections import Counter

testlist = ['red','red','blue','red','red','black','yellow','black','yellow','blue']

def count_changes(data):
    c = Counter()
    c['key'] = 40006
    for item1, item2 in zip(data, data[1:]):
        if item1 != item2:
            c[item2] += 1
    return c

print(count_changes(testlist))

输出：

Counter({'key': 40006, 'blue': 2, 'black': 2, 'yellow': 2, 'red': 1})

尚不清楚如果"key"出现在测试列表中，则正确的行为应该是什么，但是修改此代码以解决该问题很简单。