我希望能够通过另一个我已经收集到的数组(具有相同的值)来过滤我的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']]
map(lambda a: list(a), filter(lambda x: x[0] in names, cats_array))
来保持列表格式。如果你这样做,你的答案将是[['test', '', '413'], ['erd', ' ', '60'], ['Td1', 'f', '904']]
。 - Nitish