python - 按数组numpy过滤

Question

我正在尝试用我收集的另一个数组(具有相同的值)过滤我的 ndarray

我的主要 ndarray 看起来像

[['Name' 'Col1' 'Count']
 ['test' '' '413']
 ['erd' ' ' '60']
 ..., 
 ['Td1' 'f' '904']
 ['Td2' 'K' '953']
 ['Td3' 'r' '111']]

我有另一个包含各种匹配名称的列表

names = ['Td1','test','erd']

我想做什么

我想使用列表名称作为上面 ndarray 的过滤器吗？

我尝试过的

name_filter = main_ndarray[:,0] == names

这行不通

我的期望

[['Name' 'Col1' 'Count']
 ['test' '' '413']
 ['erd' ' ' '60']
 ['Td1' 'f' '904']]

Answer 1

考虑将 Pandas 用于此类数据:

import pandas as pd

data = [['Name', 'Col1', 'Count'],
        ['test', '', '413'],
        ['erd', ' ', '60'],
        ['Td1', 'f', '904'],
        ['Td2', 'K', '953'],
        ['Td3', 'r', '111']]

df = pd.DataFrame(data[1:], columns=data[0])
names = ['Td1','test','erd']
result = df[df.Name.isin(names)]

结果:

>>> df
   Name Col1 Count
0  test        413
1   erd         60
2   Td1    f   904
3   Td2    K   953
4   Td3    r   111
>>> result
   Name Col1 Count
0  test        413
1   erd         60
2   Td1    f   904
>>>

引用资料

• http://pandas.pydata.org/
• Filter dataframe rows if value in column is in a set list of values