python - 如何将 csv.Sniffer 用于 2 种不同的 CSV 类型？

Question

我想读取 2 种不同类型的 CSV 文件:

一个带 ','作为分隔符

一个带 ';'作为分隔符

我试图通过执行以下操作来检查我正在使用的分隔符:

dialect = csv.Sniffer().sniff(csvfile, [',', ';'])  
data = csv.reader(csvfile, dialect)

但后来我得到了 TypeError : expected string or buffer .

如果我这样做，它会起作用，但是我不知道何时使用什么分隔符。

data = csv.reader(csvfile, delimiter = ",")  
data = csv.reader(csvfile, delimiter = ";")

有人能帮助我吗？

Answer 1

嗅探器期望 sample string ，不是文件。您需要做的就是:

dialect = csv.Sniffer().sniff(csvfile.readline(), [',',';'])
csvfile.seek(0)  
data = csv.reader(csvfile, dialect)

查找很重要，因为您正在使用 readline 命令移动文件中的当前位置，并且您需要重置回文件的开头。否则你会丢失数据。