[英]filter by array numpy
我正在嘗試通過收集的另一個數組(具有相同的值)來過濾我的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']]
您也可以使用filter
功能。
cats_array = numpy.array(
[['Name' ,'Col1', 'Count'],
['test', '' ,'413'],
['erd' ,' ' ,'60'],
['Td1' ,'f' ,'904'],
['Td2' ,'K' ,'953'],
['Td3' ,'r', '111']]
)
names = ['Td1','test','erd']
filter(lambda x: x[0] in names, cats_array)
給出:
[array(['test', '', '413'],
dtype='|S5'), array(['erd', ' ', '60'],
dtype='|S5'), array(['Td1', 'f', '904'],
dtype='|S5')]
考慮將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
>>>
參考文獻
我也將使用@YXD的Pandas解決方案,但是出於完整性考慮,我還提供了一個基於列表理解的簡單解決方案:
data = [['Name', 'Col1', 'Count'],
['test', '', '413'],
['erd', ' ', '60'],
['Td1', 'f', '904'],
['Td2', 'K', '953'],
['Td3', 'r', '111']]
names = ['Td1', 'test', 'erd']
# select all sublist of data
res = [l for l in data if l[0] in names]
# insert the first row of data
res.insert(0, data[0])
然后為您提供所需的輸出:
[['Name', 'Col1', 'Count'],
['test', '', '413'],
['erd', ' ', '60'],
['Td1', 'f', '904']]
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.