如何从Python Numpy array1中提取特定值,并从这些值中创建与array1形状相同的新array2

时间:2020-05-21 15:57:55

标签: python arrays numpy multidimensional-array extract

假设我有一个名为“ Z”形的二维Numpy数组(90,360)。数组Z的取值范围是0到100。

如何创建一个名为“ A”的新Numpy数组,该数组仅提取特定的Z值(假设Z = 60,并将这些Z值放置在与Z中相同的A索引中)换句话说,数组'A'只需要包含等于60的Z值,并且还必须保留'Z'= 60的相同索引值。

1 个答案:

答案 0 :(得分:1)

假设数组az不是60的地方为零,我们得到:

a = np.zeros_like(z)
mask = np.isclose(z, 60)
a[mask] = z[mask]
相关问题