概述
[[10 15 1.0] [21 13 1.0] [9 14 0.0] [14 24 1.0] [21 31 0.0] ...]
我想创建两个单独的矩阵:一个包含第三列= 0.0的所有数据,另一个包含第三列的所有数据= 1.0.所以基本上将数据按第三列中的值0.0或1.0分割.
演示
>>> import numpy >>> A = numpy.array([[1,1],[2,[3,0],[4,[5,0]]) >>> A1 = A[A[:,2] == 1,:] # extract all rows with the third column 1 >>> A0 = A[A[:,2] == 0,:] # extract all rows with the third column 0 >>> A0 array([[3,0]]) >>> A1 array([[1,1]])
总结
以上是编程之家为你收集整理的Python:按列值分隔矩阵全部内容,希望文章能够帮你解决Python:按列值分隔矩阵所遇到的程序开发问题。
如果您也喜欢它,动动您的小指点个赞吧