相关疑难解决方法(0)

如何在python中的csv.reader中使用pkgutils.get_data?

我有一个python模块,其中包含各种数据文件(代表曲线的一组csv文件),需要在运行时加载。csv模块效果很好

  # curvefile = "ntc.10k.csv"
  raw = csv.reader(open(curvefile, 'rb'), delimiter=',')
Run Code Online (Sandbox Code Playgroud)

但是,如果我将此模块导入另一个脚本,则需要找到数据文件的完整路径。

/project
   /shared
       curve.py
       ntc.10k.csv
       ntc.2k5.csv
   /apps
       script.py
Run Code Online (Sandbox Code Playgroud)

我希望script.py仅按基本文件名而不是完整路径来引用曲线。在模块代码中,我可以使用:

pkgutil.get_data("curve", "ntc.10k.csv") 
Run Code Online (Sandbox Code Playgroud)

可以很好地找到文件,但是它返回已读入的csv文件,而csv.reader需要文件句柄本身。有什么方法可以使这两个模块一起正常工作?它们都是标准的库模块,因此我并不是真的期望出现问题。我知道我可以开始拆分pkgutil二进制文件数据,但是那样的话我也可能不使用csv库。

我知道我可以只在模块代码中使用它,而不必理会pkgutils,但看起来pkgutils确实正是此目的。

this_dir, this_filename = os.path.split(__file__)
DATA_PATH = os.path.join(this_dir, curvefile)
raw = csv.reader(open(DATA_PATH, "rb"))
Run Code Online (Sandbox Code Playgroud)

python

2
推荐指数
2
解决办法
3313
查看次数

标签 统计

python ×1