首页 > 解决方案 > 将数组传递给numpy中的数组

问题描述

import numpy as np

arr1= np.array([10,20,30,40,50])
arr2= np.array([False, False,  True,  True,  True])

result=arr1[arr2] #[30 40 50]

很难把握这一点。当另一个 numpy 数组传递给现有的 numpy 数组时,所有元素都相乘吗?

标签: pythonnumpy

解决方案


您正在通过另一个数组索引一个 numpy 数组,而不是一个数组传递给另一个数组。在您的情况下,由于您使用的是布尔索引,因此它正在创建一个mask,并且只返回arr1wherearr2为 True 的值。


推荐阅读