csv Python包中的内置数据集

enxuqcxy 于 2023-04-03 发布在 Python

关注(0)|答案(1)|浏览(105)

我做了一个Python包，我想上传到PyPI。我希望这个包有基于csv文件的内置数据集。包的文件夹'load'包含内置数据集所需的csv文件。
你能帮忙吗？
我做了这个代码，但它不工作

def load_data():

    file_path = pkg_resources.resource_filename('package', 'load/data.csv')
    df = pd.read_csv(file_path)
    return df

我希望调用函数load_data()来显示我的数据集。

1条答案

如果您使用setup.py，您可以添加一个package_data tgat引用您的包中的文件位置，如：

package_data = {
    '': [
        'src/myfolder/thing1.csv',
        'src/myfolder/thing2.csv',
    ],
}

更多信息here和here