在两个python numpy数组中检查条件的最快方法是什么？

Question

有什么方法可以使以下代码更快？

for q in range (1155):
  for p in range(1683):
    if original_data[q, p] == 0 and rain100[q, p] == 0:
      original_data[q, p] = np.nan
      rain100[q, p] = np.nan

在这里，我正在处理两个数组。 我认为，如果只处理一个数组，我们可以使其更快。 例如，

original_data[original_data == 0] = np.nan

我认为这比按索引运行循环索引要快得多。

如果我们处理两个数组，有没有办法做类似的事情？

Answer 1

用2个（或更多）数组定义布尔掩码很容易：

mask = (original_data == 0) & (rain100 == 0)
original_data[mask] = np.nan
rain100[mask] = np.nan