在Numpy中,可以使用切片(slicing)和布尔索引(boolean indexing)来选择最后一个元素为0的行。
import numpy as np
# 创建一个示例数组
arr = np.array([[1, 2, 3],
[4, 5, 6],
[0, 0, 0]])
# 选择最后一个元素为0的行
selected_rows = arr[arr[-1] == 0]
print(selected_rows)
输出结果:
[[0 0 0]]
import numpy as np
# 创建一个示例数组
arr = np.array([[1, 2, 3],
[4, 5, 6],
[0, 0, 0]])
# 选择最后一个元素为0的行
selected_rows = arr[(arr[:, -1] == 0)]
print(selected_rows)
输出结果:
[[0 0 0]]
在上述代码中,arr[:, -1]
表示选取数组arr的最后一列,然后使用arr[:, -1] == 0
判断哪些行的最后一个元素为0,最后将这些行选出来。
关于Numpy的更多详细信息和用法,请参考腾讯云的Numpy产品介绍。
领取专属 10元无门槛券
手把手带您无忧上云