在 python 中映射 csv

Question

我正在尝试在 python 中的 csv 文件上使用“map”。

然而，map(lambda x: x, reseller_csv)行什么也没给出。

我尝试迭代 csv 对象，它工作正常并且可以打印行。

这是代码。

# imports
import csv

# Opens files
ifile = open('C:\Users\josh.SCL\Desktop\Records.csv', 'r')
ofile = open('C:\Users\josh.SCL\Desktop\RecordsNew.csv', 'w')
resellers_file = open('C:\Users\josh.SCL\Desktop\Reseller.csv', 'r')


# Setup CSV objects
csvfile = csv.DictReader(ifile, delimiter=',')
reseller_csv = csv.DictReader(resellers_file, delimiter=',')

# Get names only in resellers
resellers = map(lambda x: x.get('Reseller'), reseller_csv)

Answer 1

csv.DictReader 是一个一次性使用的小工具。您可能已经运行了第二次。

>>> import csv
>>> iterable = ['Reseller,cost', 'fred,100', 'joe,99']
>>> reseller_csv = csv.DictReader(iterable)
>>> map(lambda x: x.get('Reseller'), reseller_csv)
['fred', 'joe']
>>> map(lambda x: x.get('Reseller'), reseller_csv)
[]
>>>

当我们在这里时：

(1) [Python 2.x] 始终以 BINARY 模式打开 csv 文件。
[Python 3.x] 始终以文本模式（默认）打开 csv 文件，并使用 newline=''

(2) 如果您坚持在 Windows 中硬编码文件路径，请使用 r"... .”而不是“...”，或使用正斜杠 - 否则 \n 和 \t 将被解释为控制字符。

Answer 2

以下内容对我有用：

>>> data = ["name,age", "john,32", "bob,45"]
>>> list(map(lambda x: x.get("name"), csv.DictReader(data))) # Python 3 so using list to see values.
['john', 'bob']

您确定从 DictReader 中获取了任何数据吗？在此之前您是否读取过任何数据，也许会让读者筋疲力尽？

Answer 3

首先针对您的具体问题：尝试检查是否确实存在名为“Reseller”的密钥，很可能它的大小写不同或有额外的空间。查看所有键的列表（假设未耗尽 DictReader）：

>>> csvfile.next().keys()

否则 map() 应该可以正常工作。但我认为这样做更具可读性（而且更快！）：

resellers = [x['Reseller'] for x in reseller_csv]