检查两个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