首页 > 解决方案 > 如何创建 Python 用户定义函数来读取 CSV 文件?

问题描述

我想创建一个函数来读取我要作为 CSV 导入的每个文件,并且我想读取的所有文件都在我桌面上的同一个文件夹中。那么,如果我希望输入是文件名,那么这段代码是否正确(因为每次我想以 CSV 格式读取文件时,它是唯一不同的东西)?

def readCSV(filename):
    filepath = 'C:\\Drive\\Desktop\\Folder\\ + filename'
    readFile = pd.read_csv(filepath)
    return readFile

还是只有在代码是这样的情况下才有效?

def readCSV(filename):
    filepath = C:\\Drive\\Desktop\\Folder\\ + filename
    readFile = pd.read_csv('filepath')
    return readFile

我对第一个函数的主要关注是文件名在一个字符串中(也就是文件路径),所以我不确定这是否适用于 Python。

标签: pythonpandascsvuser-defined-functions

解决方案


文件路径名应该是正确的字符串格式,就像您使用print()函数输出文件路径一样,它应该与您希望路径名出现的方式完全相同。

因此,您的路径应该是 2 个字符串的串联,第一个是 C: 驱动器的绝对路径,第二个是您的文件名。

"C:\\Drive\\Desktop\\Folder\\" + filename

或者,如果您想使用字符串格式:

f"C:\\Drive\\Desktop\\Folder\\{filename}"

推荐阅读