如何从 Python Numpy array1 中提取特定值并从这些值中创建一个与 array1 具有相同形状的新 array2答案

【问题标题】：How to extract specific values from Python Numpy array1 and from those values, create a new array2 with the same shape as array1如何从 Python Numpy array1 中提取特定值并从这些值中创建一个与 array1 具有相同形状的新 array2
【发布时间】：2020-09-08 07:46:47
【问题描述】：

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

如何创建一个名为“A”的新 Numpy 数组，它只提取特定的 Z 值（假设 Z=60 的位置）并将这些 Z 值放在与它在 Z 中出现的位置相同的 A 索引中. 换句话说，数组 'A' 只需要包含等于 60 的 Z 值，并且还需要保留与 where 'Z'= 60 相同的索引值。

【问题讨论】：

A 的其他元素与Z 不匹配的元素呢？
可以是零值或南值

标签： python arrays numpy multidimensional-array extract

【解决方案1】：

假设数组 a 在 z 不是 60 的位置为零，我们得到：

a = np.zeros_like(z)
mask = np.isclose(z, 60)
a[mask] = z[mask]

【讨论】：

如果z是整数类型，z == 60就可以了。我怀疑isclose 无论如何都会为您处理。