这段代码如何Pythonic [for]

问题描述

问 题

我在用plt画饼状图。如下是简化的代码。

a_part = [2001, 12000]
b_part = [1001, 2000]
c_part = [11, 1000]
d_part = [1, 10]

data = range(1, 12000)
labels = [a_part, b_part, c_part, d_part]
sizes = []
for part in labels:
    sum = 0
    for each in data:
        sum += each if each >= part[0] and each <= part[1] else 0
    sizes.append(sum)
print(sizes)

其中这段代码如何Pythonic?

for part in labels:
    sum = 0
    for each in data:
        sum += each if each >= part[0] and each <= part[1] else 0
    sizes.append(sum)

解决方案

sizes = [sum(x for x in data if low<=x<=high) for low,high in labels]

这篇关于这段代码如何Pythonic [for]的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！