【发布时间】:2018-02-18 12:33:07
【问题描述】:
有没有办法让 read_csv 自动检测分隔符? numpy 的 genfromtxt 就是这样做的。我的文件具有单空格、双空格和制表符作为分隔符的数据。 genfromtext() 解决了它,但比 pandas 的 read_csv 慢。有什么想法吗?
【问题讨论】:
标签: python pandas csv delimiter
有没有办法让 read_csv 自动检测分隔符? numpy 的 genfromtxt 就是这样做的。我的文件具有单空格、双空格和制表符作为分隔符的数据。 genfromtext() 解决了它,但比 pandas 的 read_csv 慢。有什么想法吗?
【问题讨论】:
标签: python pandas csv delimiter
另一种选择是使用内置的 CSV Sniffer。我将它与仅读取一定数量的字节混合在一起,以防 CSV 文件很大。
import csv
def get_delimiter(file_path, bytes = 4096):
sniffer = csv.Sniffer()
data = open(file_path, "r").read(bytes)
delimiter = sniffer.sniff(data).delimiter
return delimiter
【讨论】:
为了更好地控制,我使用 python 项目中名为 detect_delimiter 的 python 模块。见https://pypi.org/project/detect-delimiter/。它已经存在了一段时间。与所有代码一样,您应该在部署之前使用解释器进行测试。我已经测试到 python 版本 3.8.5。
请参见下面的代码示例,其中自动检测分隔符,而 var 分隔符是从方法的输出中定义的。然后代码读取 CSV 文件 带 sep = 分隔符。我已经使用以下分隔符进行了测试,尽管其他分隔符应该可以: , |
它不适用于多字符分隔符,例如“,”
注意!此方法不会检测格式错误的 CSV 文件。在这种情况下 其中输入文件包含两者;并且,该方法返回 ,作为检测到的分隔符。
from detect_delimiter import detect
import pandas as pd
with open(security_rule_file.csv) as myfile:
firstline = myfile.readline()
myfile.close()
deliminter = detect(firstline)
records = pd.read_csv(security_rule_file.csv, sep = delimiter)
【讨论】:
deliminter,应该写delimiter。
选项 1
使用delim_whitespace=True
df = pd.read_csv('file.csv', delim_whitespace=True)
选项 2
将正则表达式传递给sep 参数:
df = pd.read_csv('file.csv', sep='\s+')
这相当于第一个选项
【讨论】:
sep='\t'。