假设我有一个名为“ Z”形的二维Numpy数组(90,360)。数组Z的取值范围是0到100。
如何创建一个名为“ A”的新Numpy数组,该数组仅提取特定的Z值(假设Z = 60,并将这些Z值放置在与Z中相同的A索引中)换句话说,数组'A'只需要包含等于60的Z值,并且还必须保留'Z'= 60的相同索引值。
答案 0 :(得分:1)
假设数组a
在z
不是60的地方为零,我们得到:
a = np.zeros_like(z)
mask = np.isclose(z, 60)
a[mask] = z[mask]