您好, 欢迎来到 !    登录 | 注册 | | 设为首页 | 收藏本站

Python:按列值分隔矩阵

5b51 2022/1/14 8:22:49 python 字数 979 阅读 616 来源 www.jb51.cc/python

我有一个矩阵A有3列看起来像,但更大: [[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分割. 如果您使用的是 Numpy,请首先找到第三列具有所需值的行,然后使用

概述

[[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:按列值分隔矩阵所遇到的程序开发问题。


如果您也喜欢它,动动您的小指点个赞吧

除非注明,文章均由 laddyq.com 整理发布,欢迎转载。

转载请注明:
链接:http://laddyq.com
来源:laddyq.com
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。


联系我
置顶