在嵌套列表中搜索元素并将其打印出来

时间:2018-07-24 14:32:43

标签: python python-3.x

我有一个看起来像这样的嵌套列表

[['ID', 'Name'], ['E001', 'Marcus Tan'], ['E002', 'Mary Tay'], ['E003', 'Patrick Goh'], ['E004', 'Joey Lim'], ['E005', 'Edward Lim']]

我必须搜索他们名字的一部分,例如('Ma')

假定的结果应该是

Enter any part of name: Ma

Emp ID  Name 

E001    Marcus Tan 

E002    Mary Tay

到目前为止,这是我的代码

def search_emp():
    name = input('Enter any part of name: ')
    emp_list = []
    print('{:<5} {:>7}'.format('Emp ID', 'Name'))
    print('-------- -----------------')
    with open('Assignment_Data1.csv') as f:
        if name in emplist:
            print('{}'.format(name))
        else:
            print('Sorry there are no such employees.')

2 个答案:

答案 0 :(得分:2)

对于很多行,可以使用pandas非常容易可视化数据;)

import pandas as pd

>>> df = pd.DataFrame(values, columns=['ID','name'])
>>> df[df.name.str.contains('Ma')]

    ID  name
1   E001    Marcus Tan
2   E002    Mary Tay

答案 1 :(得分:1)

尝试一下:

echo