将条件应用于多个数组

Question

如何将相同条件应用于多个数组？

set /?

像这样，我想使用相同的过滤器过滤掉“ a”和“ b”中的元素，然后替换原始值。尽管这段代码可以正常工作，但是如果我想将此过滤器应用于许多数组，将会很长。

Answer 1

您可以使用*args来引用所有参数并立即返回所有结果数组。 for arg in args[:-1]循环到倒数第二个参数（在您的情况下为b）。

def check(*args):
    return [arg[args[-1]>6] for arg in args[:-1]] # args[-1] is c in your case

输出 您可以访问返回列表中的元素作为所需的输出。

[array([3, 4, 5]), array([4, 5, 6])]