python - Numpy ndarray 在 bool 条件之后进行广播

Question

我想知道是否有一种方法可以通过避免使用 for 循环来利用 python numpy 数组广播来减少计算时间。这是以下最小示例:

import numpy as np

# parameters
n_t = 256
G = 0.5
k_n = 10

# typical data
tau = np.linspace(0,2*np.pi,256)
x_t = np.sin(tau).reshape((n_t,1))

delta = np.maximum(0,(x_t-G))
f_dot = np.zeros((n_t,1))

for i  in range(0,n_t,1):
    # boolean condition
    if delta[i,0] > 0:
        f_dot[i,0] = k_n

如有任何建议，我们将不胜感激。谢谢。

Answer 1

您可以使用np.where为了根据条件的结果从 k_n 或 f_dot 分配值:

f_dot = np.where(delta > 0, k_n, f_dot)