V2EX = way to explore
V2EX 是一个关于分享和探索的地方
现在注册
已注册用户请  登录
爱意满满的作品展示区。
YUX
V2EX  ›  分享创造

写了一个优化 pandas 内存的工具

  •  2
     
  •   YUX ·
    YUX · 2020-05-04 21:11:44 +08:00 · 2533 次点击
    这是一个创建于 1476 天前的主题,其中的信息可能已经有所发展或是发生改变。

    pandas 爆内存主要是两个原因

    • 默认 int64, float64
    • object 类型经常有太多重复项

    针对这两点都做了优化, int 、float 转化为合适都类型, object 如果重复项太多就转化为 category

    用法很简单

    import fast_csv as fc
    data = fc.read_csv('$PATH/$FILE.csv')
    

    一般来讲 (我最近用的几个 kaggle 的 data),能减少 50%内存,有时能减少 90%以上. 没有什么副作用,和一般直接用pandas读 csv 是一样的,都是 pandas.DataFrame

    开源在 https://github.com/YUX-IO/fast_csv

    安装 pip install fast_csv

    4 条回复    2020-05-04 22:17:25 +08:00
    lithiumii
        1
    lithiumii  
       2020-05-04 21:18:20 +08:00 via Android
    kaggle
    nzd
        2
    nzd  
       2020-05-04 21:44:49 +08:00
    这个有点厉害,学习了
    dlsflh
        3
    dlsflh  
       2020-05-04 22:03:01 +08:00 via Android   ❤️ 1
    等我上班后就实测对大量的工业传感器数据的优化效果。
    ipwx
        4
    ipwx  
       2020-05-04 22:17:25 +08:00   ❤️ 1
    。。。你需要的是在调用 pd.read_csv 的时候手动指定 dtype={...},然后 engine='C'
    关于   ·   帮助文档   ·   博客   ·   API   ·   FAQ   ·   实用小工具   ·   1377 人在线   最高记录 6679   ·     Select Language
    创意工作者们的社区
    World is powered by solitude
    VERSION: 3.9.8.5 · 27ms · UTC 17:12 · PVG 01:12 · LAX 10:12 · JFK 13:12
    Developed with CodeLauncher
    ♥ Do have faith in what you're doing.