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

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

Answer 2

试试这个：

a=[['ID', 'Name'], ['E001', 'Marcus Tan'], ['E002', 'Mary Tay'], ['E003', 'Patrick Goh'], ['E004', 'Joey Lim'], ['E005', 'Edward Lim']]
b=input("Enter any part of name: ")
print("""\nEmp ID  Name """)
[print('   '.join(e)) for e in a if b in e[1]]