要获取二维Numpy数组中每一行的前两个值的索引,但排除特定区域,可以使用以下方法:
import numpy as np
arr = np.array([[1, 2, 3, 4],
[5, 6, 7, 8],
[9, 10, 11, 12]])
indices = arr[:, :2]
这将返回一个新的数组,其中包含原数组每一行的前两个值。
excluded_rows = [0, 2]
filtered_indices = indices[~np.isin(np.arange(len(arr)), excluded_rows)]
这将返回一个新的数组,其中包含排除了特定行的索引。
综上所述,获取二维Numpy数组中每一行的前两个值的索引,但排除特定区域的完整代码如下:
import numpy as np
arr = np.array([[1, 2, 3, 4],
[5, 6, 7, 8],
[9, 10, 11, 12]])
indices = arr[:, :2]
excluded_rows = [0, 2]
filtered_indices = indices[~np.isin(np.arange(len(arr)), excluded_rows)]
print(filtered_indices)
这将输出:
[[2]
[6]]
这个结果表示原数组中第三行和第七行的前两个值的索引。
领取专属 10元无门槛券
手把手带您无忧上云