我的函数如何将csv文件作为输入？

Question

我需要将一个csv文件作为函数参数，并作为示例代码：

import openpyxl
from openpyxl import load_workbook
import csv
import xlrd


def supermarket(filename):
dictionary = {}
with open(filename, newline='') as filename:
    read_csv = csv.reader(filename, delimiter=',')
    for row in read_csv:
        dictionary += row[0]
        dictionary[row[0]] = row[1]
return dictionary

print(supermarket(prices1.csv))

但是，当我调试代码时，回溯错误表明这是一个NameError

prices1

未定义。我的教授告诉我，我的论点应该可以把prices1.csv作为一个论点，但我不明白这怎么可能。在

Answer 1

您的代码实际上引用了变量prices1的成员csv，该成员显然不存在。文件名实际上应该是一个字符串。因此您的代码可以更改为以下内容（假设csv文件与您的工作目录位于同一目录中）：

import openpyxl
from openpyxl import load_workbook
import csv
import xlrd


def supermarket(filename):
    dictionary = {}
    with open(filename, newline='') as filename:
        read_csv = csv.reader(filename, delimiter=',')
        for row in read_csv:
            dictionary += row[0]
            dictionary[row[0]] = row[1]
    return dictionary

print(supermarket("prices1.csv"))

Answer 2

如果方便的话，可以使用pandas库以更有效的方式处理CSV文件。在