Python – 计算 % K 个元素
通过条件检查数字/元素是一个常见的问题,几乎在每个程序中都会遇到。有时我们还需要获取与特定条件匹配的总数,以区分哪些不匹配以供进一步使用。让我们讨论可以实现检查可被 K 整除的数的任务的某些方法。
方法 #1:使用sum()
+ 生成器表达式
此方法使用每当生成器表达式返回 true 时将总和加 1 的技巧。当列表用尽时,返回匹配 % K 的数字计数的总和。
# Python 3 code to demonstrate
# Count % K elements
# using sum() + generator expression
# initializing list
test_list = [3, 5, 1, 6, 7, 9]
# printing original list
print ("The original list is : " + str(test_list))
# initializing K
K = 3
# using sum() + generator expression
# to get count of elements matching condition
# Count % K elements
res = sum(1 for i in test_list if i % K != 0)
# printing result
print ("The number of % K elements: " + str(res))
输出 :
The original list is : [3, 5, 1, 6, 7, 9]
The number of % K elements: 3
方法#2:使用sum() + map()
map() 的任务几乎与生成器表达式相似,不同之处只是它采用的内部数据结构不同,因此效率更高。
# Python 3 code to demonstrate
# Count % K elements
# using sum()+ map()
# initializing list
test_list = [3, 5, 1, 6, 7, 9]
# printing original list
print ("The original list is : " + str(test_list))
# initializing K
K = 3
# using sum()+ map()
# to get count of elements matching condition
# Count % K elements
res = sum(map(lambda i: i % K != 0, test_list))
# printing result
print ("The number of % K elements: " + str(res))
输出 :
The original list is : [3, 5, 1, 6, 7, 9]
The number of % K elements: 3