我有一个看起来像这样的嵌套列表
[['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.')
答案 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