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

Question

我有一个如下所示的嵌套列表

[['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.')

Answer 1

对于很多行，可以使用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