博客

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

  • Post category:Python

查找NumPy数组中一个序列的出现次数可以通过NumPy库中的count_nonzero函数实现。

下面是查找一个长度为3的序列在一个一维数组中出现次数的完整攻略：

步骤1：导入NumPy库

import numpy as np

步骤2：定义数组和序列

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

步骤3：使用count_nonzero函数进行查找

count = np.count_nonzero(np.correlate(arr, seq, mode='valid') == np.sum(seq ** 2))
print(count)

以上代码运行结果是3，表示在arr数组中，序列[1,2,1]出现了3次。具体说明如下：

  • 在第1个位置上，[1,2,1]与[1,2,3]部分匹配，不满足完全匹配条件，故不计入结果中。
  • 在第4个位置上，[1,2,1]与[1,2,1]完全匹配，计入结果中。
  • 在第6个位置上，[1,2,1]与[1,2,1]完全匹配，计入结果中。
  • 在第9个位置上，[1,2,1]与[1,2,1]完全匹配，计入结果中。

以下是另外一个例子，展示如何在多维数组中查找一个序列的出现次数：

步骤1：导入NumPy库

import numpy as np

步骤2：定义数组和序列

arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9], [1, 2, 3]])
seq = np.array([1, 2, 3])

步骤3：使用count_nonzero函数进行查找

count = np.count_nonzero(np.apply_along_axis(lambda x: np.array_equal(x, seq), 1, arr))
print(count)

以上代码运行结果是2，表示在arr数组中，序列[1,2,3]出现了2次。具体说明如下：

  • 在第1个子数组中，[1,2,3]与[1,2,3]完全匹配，计入结果中。
  • 在第4个子数组中，[1,2,3]与[1,2,3]完全匹配，计入结果中。
Tags: