如何统计二维数组指定行内特定元素的出现次数

实现方案 #

你可以直接使用Python列表内置的count()方法完成统计，指定要统计的行对应的子数组，调用count(目标元素)即可直接返回对应元素的出现次数。

对应你的示例场景，代码实现如下：

# 示例二维数组
counts = [[0,10,4,2], [0,0,0,0,5,7]]

# 统计第0行中0的出现次数
print(counts[0].count(0))
# 输出结果：1

# 统计第1行中0的出现次数
print(counts[1].count(0))
# 输出结果：4

如果需要频繁使用该统计逻辑，可以封装为通用函数，同时增加下标合法性校验避免报错：

def count_target_in_row(arr: list, row_idx: int, target) -> int:
    if row_idx < 0 or row_idx >= len(arr):
        return 0
    return arr[row_idx].count(target)

# 调用测试
print(count_target_in_row(counts, 0, 0)) # 1
print(count_target_in_row(counts, 1, 0)) # 4

内容的提问来源于stack exchange，提问作者Saad Sarawan