Python一行代码统计数组元素个数：利用count方法

想法由来

想法由来

在编写代码时，我遇到了一个具体需求：需要统计一个列表中值为1的元素数量。该列表中的元素仅包含0或1两种数值。

起初，我考虑编写一个独立的函数来完成这一任务。

# 返回一个0,1数组中1的数量
def num_one(source_array):
 count = 0
 for x in source_array:
  if x == 1:
   count += 1
 return count

随后，我认为这种传统方式不够简洁，于是探索Python是否支持通过一行代码实现相同功能，最终发现确实可行。

count方法描述

count方法描述

这里使用了Python内置的count()方法，该方法用于统计字符串中指定子字符串出现的次数。可选参数用于指定搜索的起始和结束位置。

count方法语法与参数

count方法语法与参数

语法：

str.count(sub, start= 0,end=len(string))

参数：

- sub – 搜索的子字符串

- start – 字符串开始搜索的位置。默认为第一个字符,第一个字符索引值为0。

- end – 字符串中结束搜索的位置。字符中第一个字符的索引为 0。默认为字符串的最后一个位置。

具体实现

具体实现

实现思路是将列表转换为字符串，然后使用count方法计算目标元素的数量。代码如下：

num = str(source_array).count("1")

如果源数组是numpy.array类型，则需先将其转换为list，再进行计数操作：

num = str(source_array.tolist()).count("1")

以上这篇用python一行代码得到数组中某个元素的个数方法就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持本站。