Python-如果行包含则删除矩阵/数组中的行

Question

如果你有一个 x*n 矩阵，如何检查包含特定数字的行，如果是，如何删除该行？

Answer 1

如果您使用pandas，您可以创建一个可用于索引数据帧的掩码，并使用~否定该掩码:

df = pd.DataFrame(np.arange(12).reshape(3, 4))
#    0  1   2   3
# 0  0  1   2   3
# 1  4  5   6   7
# 2  8  9  10  11

value = 2

如果您想检查该值是否包含在特定列中:

df[~(df[2] == value)]
#    0  1   2   3
# 1  4  5   6   7
# 2  8  9  10  11

或者如果它可以包含在任何列中:

df[~(df == value).any(axis=1)]
#    0  1   2   3
# 1  4  5   6   7
# 2  8  9  10  11

之后只需将其重新分配给df即可。

<小时/>

如果您仅使用 numpy，这也适用:

x = np.arange(12).reshape(3, 4)
# array([[ 0,  1,  2,  3],
#        [ 4,  5,  6,  7],
#        [ 8,  9, 10, 11]])

x[~(x == value).any(axis=1)]
# array([[ 4,  5,  6,  7],
#        [ 8,  9, 10, 11]])

<小时/>

最后，如果您使用普通 Python 并且有一个列表列表，请在列表理解中使用内置 any:

y = [[0, 1, 2, 3], [4, 5, 6, 7], [8, 9, 10, 11]]

[row for row in y if not any(x == value for x in row)]
# [[4, 5, 6, 7], [8, 9, 10, 11]]