基于目标条件的数组元素的随机抽样

以下是基于目标条件的数组元素的随机抽样的一个解决方法的示例代码：

import random

def random_sample_with_condition(arr, condition, k):
    # 使用列表推导式筛选满足条件的元素
    filtered_arr = [x for x in arr if condition(x)]
    
    # 如果筛选后的元素个数小于 k，则返回全部元素
    if len(filtered_arr) <= k:
        return filtered_arr
    
    # 使用 random.sample 函数随机抽样 k 个元素
    return random.sample(filtered_arr, k)

# 示例用法
arr = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]

# 定义一个条件函数，筛选出大于等于 5 的元素
def condition(x):
    return x >= 5

k = 3

sampled_arr = random_sample_with_condition(arr, condition, k)
print(sampled_arr)

这段代码首先定义了一个 random_sample_with_condition 函数，它接受三个参数：原始数组 arr、条件函数 condition 和抽样数量 k。该函数首先使用列表推导式筛选出满足条件的元素，然后判断筛选后的元素个数是否小于 k，如果是，则直接返回全部元素；如果不是，则使用 random.sample 函数从筛选后的元素中随机抽样 k 个元素，并返回抽样结果。

在示例用法中，定义了一个条件函数 condition，它筛选出大于等于 5 的元素。然后调用 random_sample_with_condition 函数，传入原始数组 arr、条件函数 condition 和抽样数量 k，得到抽样结果并打印输出。