numpy是一个开源的Python科学计算库,提供了高效的多维数组对象和各种用于数组操作的函数。要获取numpy数组中某些列中元素为零的行索引,可以使用以下方法:
import numpy as np
arr = np.array([[1, 0, 3],
[0, 0, 0],
[4, 0, 6],
[0, 7, 0]])
zero_rows = np.where((arr[:, 1] == 0) & (arr[:, 2] == 0))[0]
上述代码中,arr[:, 1]
表示获取数组arr的第2列,arr[:, 2]
表示获取数组arr的第3列。通过使用逻辑与运算符&
,可以筛选出第2列和第3列元素都为零的行。np.where
函数返回满足条件的行索引。
print(zero_rows)
输出结果为:
[1]
上述结果表示在数组arr中,第2行满足条件(第2列和第3列元素都为零)。
推荐的腾讯云相关产品:腾讯云服务器(CVM)和腾讯云对象存储(COS)。
领取专属 10元无门槛券
手把手带您无忧上云