python - 如何通过元素长度清除 numpy 数组

Question

我有一个二维 numpy 数组:

a=np.array([[1,2,3], 
            [4,5,6,7], 
            [2,3,4]])

如何在不知道其位置的情况下有效删除包含 3 个以上元素(列)的元素(行)？

    a=np.array([[1,2,3], 
                [2,3,4]])

“删除所有长度超过 3 的行。”

Answer 1

这应该可以解决问题(假设数组在变量 a 中):

np.array([row for row in a if len(row)<=3])

我在 numpy 数组 a 上使用列表推导式，只选择长度为 3 或更小的元素。