对于带有0和1的Numpy数组,可以使用以下方法删除重复为1的行:
完整的代码如下:
import numpy as np
arr = np.array([[0, 1, 0],
[1, 1, 0],
[1, 1, 1],
[0, 1, 1],
[1, 0, 0]])
unique_rows = np.unique(arr, axis=0)
duplicate_rows = np.logical_and(unique_rows[:, 1:].sum(axis=1) > 0, unique_rows[:, 0] == 1)
filtered_rows = unique_rows[~duplicate_rows]
print(filtered_rows)
输出结果为:
[[0 1 0]
[1 0 0]]
这个代码的作用是删除重复为1的行,保留唯一的行。首先,使用np.unique函数找到数组中唯一的行。然后,创建一个布尔索引数组,用于标记重复为1的行。最后,使用布尔索引数组删除重复为1的行,得到最终的结果。
推荐的腾讯云相关产品:腾讯云服务器(CVM)和腾讯云云函数(SCF)。
领取专属 10元无门槛券
手把手带您无忧上云