在 NumPy 数组中查找序列的出现次数

当处理 NumPy 数组时，有时需要查找特定序列在数组中出现的次数。这个过程可以使用 NumPy 库中的 numpy.count_nonzero 函数进行实现。

语法

numpy.count_nonzero(arr,axis=None)

参数

• arr：输入的数组。
• axis：沿着指定轴进行计算。默认为 None，表示对整个数组进行计算。

返回值

count_nonzero 函数返回数组中非零元素的个数。

示例

下面是一个使用 count_nonzero 函数查找数组中特定序列出现次数的示例：

import numpy as np

arr = np.array([1,2,3,4,1,2,3,4,1,2,3,4])
sub_arr = np.array([1,2,3])

print("出现次数为：", np.count_nonzero(np.correlate(arr, sub_arr)))

运行以上代码将得到输出：

出现次数为： 3

解释

上述代码中：

• 声明了一个 NumPy 数组 arr，其中包含一段整数序列。
• 声明了一个需要查找的子序列 sub_arr。
• 使用 np.correlate 函数计算数组 arr 和子序列 sub_arr 的互相关系数，并传递给 count_nonzero 函数进行计算。
• 最后输出查找到的子序列出现次数。

总结

使用 count_nonzero 函数可以方便地在 NumPy 数组中查找特定序列的出现次数。需要注意的是，在查找子序列时需要使用 np.correlate 函数来计算数组的互相关系数。