我有Xy：
matrix([[1, 1, 1, 1],
[1, 2, 2, 2],
[2, 3, 3, 3],
[2, 4, 4, 4]])
如何返回第一列== 1的所有行？
答案应该是：
matrix([[1, 1, 1, 1],
[1, 2, 2, 2]])
该matrix类型通常应该有利于避免numpy.array。但是，如果您觉得需要使用它，则可以使用过滤矩阵numpy.compress。
这样做时，平整滤镜很重要...
>>> numpy.compress((m[:,0] == 1).flat, m, axis=0)
matrix([[1, 1, 1, 1],
[1, 2, 2, 2]])
在这里，m[:,0] == 1返回一个布尔矩阵，指示哪些行被认为是可接受的。numpy.compress会为我们选择那些可接受的行。
本文收集自互联网，转载请注明来源。
如有侵权，请联系 [email protected] 删除。
我来说两句