python - 使用 numpy 数组作为 lambda 参数？

Question

是否有一种合理的方法可以在一行中完成以下操作？我真的很想避免创建临时变量或单独的函数。

import numpy as np
x = np.array([1,2,3,4,5])
x = np.ma.masked_where(x>2, x)

我试过了

x = map(lambda x: np.ma.masked_where(x>2, x), np.array([1,2,3,4,5]))

但是 map 对象不是我想要的？我当然可以定义单独的函数，从而避免分配变量:

masker = lambda x: np.ma.masked_where(x>2, x)
x = masker(np.array([1,2,3,4,5]))

Answer 1

你根本不需要map，只需要一个匿名函数。您要做的就是用函数调用中的参数绑定(bind)替换对 x 的初始赋值。

import numpy as np
# x = np.array([1,2,3,4,5])
# x = np.ma.masked_where(x>2, x)

x = (lambda x: np.ma.masked_where(x>2, x))(np.array([1,2,3,4,5]))