在使用2D布尔掩码时,可以通过以下步骤从2D numpy数组中按行获取所有第一个值:
import numpy as np
arr = np.random.randint(0, 10, size=(5, 5))
mask = arr > 5
first_values = np.take(arr, np.argmax(mask, axis=1))
在上述代码中,np.argmax(mask, axis=1)
返回每行中第一个满足条件的元素的索引,然后np.take(arr, ...)
根据索引获取对应的值。
完整的代码示例:
import numpy as np
arr = np.random.randint(0, 10, size=(5, 5))
mask = arr > 5
first_values = np.take(arr, np.argmax(mask, axis=1))
print("2D数组:")
print(arr)
print("\n布尔掩码:")
print(mask)
print("\n每行第一个满足条件的值:")
print(first_values)
这样,你就可以从2D numpy数组中按行获取所有第一个值了。
对于腾讯云相关产品和产品介绍链接地址,由于要求不能提及具体的云计算品牌商,无法提供相关链接。但是,腾讯云提供了丰富的云计算服务,包括云服务器、云数据库、云存储等,可以根据具体需求选择适合的产品。
没有搜到相关的沙龙
领取专属 10元无门槛券
手把手带您无忧上云