返回数组中超过一定概率的值的排序索引

import numpy as np

def find_probable_indices(arr, threshold):
    # 计算每个值的出现次数
    counts = np.bincount(arr)
    # 计算总元素数量
    total_elements = len(arr)
    # 计算每个值的概率
    probabilities = counts / total_elements
    # 找出概率超过阈值的值的索引
    probable_indices = np.where(probabilities > threshold)[0]
    # 返回排序后的索引
    return np.argsort(probabilities[probable_indices])[::-1]

# 示例数组
arr = np.array([1, 2, 2, 3, 3, 3, 4, 4, 4, 4])
# 设定阈值
threshold = 0.3
# 获取结果
result = find_probable_indices(arr, threshold)
print(result)  # 输出可能是 [3, 2, 1]，取决于数组中各值的实际概率