NumPy ndArray中基于布尔值的最长序列求解方法

import numpy as np

# 创建一个示例数组
arr = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9, 10])

# 定义条件，例如判断数组中的元素是否大于5
condition = arr > 5

# 使用连续布尔值的方法，找到最长的连续True序列
longest_sequence = np.diff(np.where(np.concatenate(([condition[0]], condition[:-1] != condition[1:], [True])))[0])[::2].max()

# 根据最长连续True序列的起始位置和长度，得到最长序列的索引范围
start_index = np.where(condition)[0][np.argmax(np.diff(np.where(np.concatenate(([condition[0]], condition[:-1] != condition[1:], [True])))[0])[::2])]
end_index = start_index + longest_sequence

# 输出结果
print("最长序列的起始位置：", start_index)
print("最长序列的结束位置：", end_index)
print("最长序列的长度：", longest_sequence)
print("最长序列：", arr[start_index:end_index+1])