python 矩阵操作， 筛选符合条件的行

Posted 2023-04-21

python 矩阵操作， 筛选符合条件的行，有类似于matlab那样find函数功能的函数吗？

我举个简单的例子：

取出含有元素0的所有行

import numpy as np 
x = np.array([[1,2,3,4,0],[2,3,4,5,6],[0,1,2,3,4]])
b=[]
for row in x:
    for i in row:
        if i==0:
            b.append(row)
print b

PS G:\\Python learning-Q> python exbaidu.py

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

参考技术A python的第三方库numpy（用于矩阵运算，需要import numpy as np）中可以使用np.where，如
>>a = np.array(a)
>>a
array([1, 2, 3, 1, 2, 3, 1, 2, 3])
>>idx = np.where(a > 2)
>>idx
(array([2, 5, 8], dtype=int32),) 参考技术B #使用pandas模块的data frame，这个类似于R或matlab下的数据框，其他功能也类似

import pandas as pd

read_data = pd.DataFrame('file')

#然后在此基础上操作吧
#如果只使用numpy，有np.where函数，可以去看一下help文档吧

Notaped++标记选择符合条件的行书签

一个文本中有很多行文本，但是只是需要部分，我们就可以用查找(Ctrl+F)--标记来先来筛选。

文中是写了正则匹配纯数字开头的行。

来源：https://www.t00ls.net/viewthread.php?tid=39884&page=1&extra=#pid658558 4楼回答

本文出自 “悟透的杂货铺” 博客，请务必保留此出处http://wutou.blog.51cto.com/615096/1926337