如何从字典中提取以仅打印某些变量python

Question

我有一个像这样的 tsv 文件

Name    School   Course         
Nicole  UVA      Biology
Jenna   GWU      CS

从那里，我只想打印该字典中的名称和课程。 我该怎么办？

下面的代码是我如何将原始 TSV 文件放入上面的字典中。

import csv
data = csv.reader(open('data.tsv'),delimiter='\t')
fields = data.next()
for row in data:
    item = dict(zip(fields, row))
    print item 

所以现在我得到了这样的字典：

{'Name':'Nicole.', 'School':'UVA.','Course':'Biology'}
{'Name':'Jenna.', 'School':'GWU','Course':'CS'}
{'Name':'Shan', 'School':'Columbia','Course':'Astronomy'}
{'Name':'BILL', 'School':'UMD.','Course':'Algebra'}

我只想打印该字典中的名称和课程。 我该怎么办？

我想添加代码以便我只打印

{'Name':'Jenna.','Course':'CS'}
{'Name':'Shan','Course':'Astronomy'}
{'Name':'BILL','Course':'Algebra'}

请指导。 谢谢你

Answer 1

只需删除循环中的密钥

for row in data:
    item = dict(zip(fields, row))
    del item['School']
    print item 

Answer 2

最简单的方法是在打印之前删除item['School']条目

for row in data:
    item = dict(zip(fields, row))
    del item['School']
    print item 

但这仅在您完全知道词典的外观，没有其他不需要的条目并且已经有School条目的情况下才有效。 相反，我建议您在旧词典的基础上再建一个新词典，只保留Name和Course条目

for row in data:
    item = dict(zip(fields, row))
    item = {k, v for k, v in item.items() if k in ('Name', 'Course')}
    print item 

Answer 3

也许首先使用DictReader ，并且仅在键与预定义列表匹配时才重新DictReader行字典：

import csv

keep = {"Name","Course"}
data = csv.DictReader(open('data.tsv'),delimiter='\t')

for row in data:
    row = {k:v for k,v in row.items() if k in keep}
    print(row)

结果：

{'Course': 'Biology', 'Name': 'Nicole'}
{'Course': 'CS', 'Name': 'Jenna'}

Answer 4

根据这里的答案： 过滤python字典中的项，其中键包含特定的字符串

print {k:v for k,v in item.iteritems() if "Name" in k or "Course" in k}

Answer 5

最好使用专为此类任务（熊猫）设计的库。 字典非常适​​合存储键值对，但是看起来您具有类似电子表格的表格数据，因此您应该选择一种存储类型，以更好地反映手头的数据。 您可以简单地执行以下操作：

import pandas as pd

df = pd.read_csv('myFile.csv', sep = '\t')

print df[['Name','Course']]

您会发现，当您开始执行更复杂的任务时，最好使用编写良好的库而不是将某些东西组合在一起

Answer 6

用下面的行替换你的“print(item)”行。

print(dict(filter(lambda e: e[0]!='School', item)))

输出：

{'Name':'Jenna.','Course':'CS'}
{'Name':'Shan','Course':'Astronomy'}
{'Name':'BILL','Course':'Algebra'}