根据条件从 numpy 数组的子数组中删除行

Question

我有以下 numpy 数组：

x = [[1,2],[3,4],[10,1]]
y = [[5,6],[1,8],[7,8]]
z = [[10,2],[9,10],[11,12]]
xyz = np.array([x,y,z])

我想在xyz中的x 、 y 、 z中的每一个的第一列中删除值为 10 的行。 所以我想要的output：

array([[[ 1,  2],
        [ 3,  4]],

       [[ 5,  6],
        [ 1,  8],
        [ 7,  8]],

       [[ 9, 10],
        [11, 12]]], dtype=object)

我试过xyz[xyz[:,:,0]!=10]但这并不能保留xyz的 3 维性质。 我想我可以将第一个维度、切片和 append 迭代到一个新数组，但我正在寻找一个更简单（可能是单线）的解决方案。

Answer 1

尝试：

np.array([a[a[:,0]!=10] for a in xyz])

但是，由于尺寸不匹配，这不再是真正的 3D numpy 数组。