Question

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

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

Answer 1

我纤兄燃毁虚举个简单的例子：

取出尘轿含有元素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])]

Answer 2

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),)

Answer 3

#使用pandas模块的data frame，这个类似于R或matlab下的数据框，其他功能也类似谨梁

import pandas as pd

read_data = pd.DataFrame('file')

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