如何从Python Numpy数组1中提取特定的值,并从这些值中创建一个与数组1形状相同的新数组2。

问题描述 投票:0回答:1

比方说,我有一个名为'Z'形(90,360)的2D Numpy数组。数组Z的值范围是0到100。

我如何创建一个新的Numpy数组,称为'A',只提取特定的Z值(比如说Z=60),并将这些Z值放置在Z中出现的相同的A-指数中,换句话说,数组'A'需要只包含等于60的Z值,并保留'Z'=60的相同指数值。

python arrays numpy multidimensional-array extract
1个回答
1
投票

假设数组 a 的地方将为零。z 是不60我们得到。

a = np.zeros_like(z)
mask = np.isclose(z, 60)
a[mask] = z[mask]
© www.soinside.com 2019 - 2024. All rights reserved.