Pandas Should Go Extinct

(eddie.codes)

98 points | by __eddie__ 2 hours ago

28 comments

  • minimaxir 2 hours ago
    It's been a while since I've seen an actual data science post submitted to Hacker News: both because AI has superset a lot of DS tasks (e.g. vector embeddings), but also because not much new has happened in DS. Polars has been around for a bit and as noted it is much better than pandas, but otherwise the DS ecosystem has been somewhat stagnant.

    I'd write more tutorials about how to use data science tooling but one consequence of AI is that all the old data sources I used to analyze such as social media and Reddit are now completely locked down (I am surprised NYC Taxi is still being updated, though). Therefore in the meantime, I'm working on making better data science tooling...although unclear to what end due to the data issue above.

  • mulmboy 11 minutes ago
    People are usually surprised to hear that polars can be slower for fairly pedestrian operations, especially with smaller datasets. For example take a 5000 x 3 dataframe of float64 and sort by one column and you'll find polars takes about 2.5x as long. If you set POLARS_MAX_THREADS to 1 then it's faster. Though this all depends on the machine. Polars tends to shine with larger datasets or where it can heavily take advantage of query planning.

    Don't skip your profiling

  • __mharrison__ 25 minutes ago
    I'm in the middle of wrapping up the edits for Effective Pandas 3rd Edition. (I also wrote a Polars book and just wrapped up a weeklong training session on pandas this week.)

    Pandas is not perfect, it has a bunch of warts. But it is good enough for most. (And many of those folks are using Excel or tableau or power bi... These were the types I was training this week).

    If you have medium data, migrating from pyarrow backed pandas to duck or Polars is trivial.

  • rmunn 12 minutes ago
    Cute title; I thought I was about to read a contrarian ecologist saying "a species that is so very specialized in its diet and finicky in its reproductive behavior doesn't deserve, evolutionarily speaking, to survive". (Seriously, it's seriously freakin' difficult to get pandas to reproduce in captivity). And while my knee-jerk reaction would be "maybe, but we should still preserve them because we can", I was prepared to see if the author had a serious argument to present.

    Instead it's a cute bait-and-switch title, and the article tells you upfront that it's actually about the Python `pandas` library. Which I think I've encountered maybe once in my entire career (I'm not in the data-science field), so I don't have much meaningful to say about the article itself. I just want to commend the author on fooling me with the title. This is the kind of "clickbait" I can respect and actually wish there was a little bit more of sometimes. A nice chuckle, then a real article.

  • sjtrny 2 hours ago
    > People typically start with Excel and graduate to Pandas somewhere in the GB range. Pandas serves them well into the 10s of GBs range, and then they start hitting memory issues, slow computation, or become frustrated with Pandas’ baroque API.

    Assumes that a project moves beyond 10s of GBs. I guess 99.9% of projects that import pandas fall well below this threshold.

    • __mharrison__ 22 minutes ago
      Most folks just need to learn how to use pandas well and that will open enough doors. Then they can move to polars or duck if needed.
    • __eddie__ 2 hours ago
      True, but also if Polars and DuckDB offer a similar experience with the ability to scale beyond that range, why not use them (for new projects)?

      This isn't a call to arms to rewrite everything in the new shiny, just consider the new shiny for new shiny things

      • appplication 1 hour ago
        Not to mention less footguns. I used to spend my days unwinding bad habits DS pick up from years of panda abuse.

        Credit to pandas for popularizing dataframes in Python, but polars and duckdb are objectively better APIs in addition to their implementation improvements. Agree that it’s time we let it go.

        • __eddie__ 1 hour ago
          Yes, this was something I emphasised more in the slides / my actual talk. But it's just significantly easier to reason about the APIs for both DuckDB and Polars
    • minimaxir 2 hours ago
      At my work I had convinced the ML pipeline engineers to switch from pandas to polars for even small ETL pipelines and there were notable performance gain with better CPU/memory utilization.

      If a library is performant at large datasets, it is likely performant at small ones too.

      • sjtrny 2 hours ago
        I’m not disagreeing with that statement at all. You missed my point that there are thousands of people making small Python scripts for education and personal projects everyday. In those circumstances the performance concerns are irrelevant and the ergonomics of good pandas documentation and community knowledge make it a better choice.
        • minimaxir 1 hour ago
          That inertia is not a good thing, and it's partially why there's stagnation in data science. Polars is more than mature enough in both documentation and resources for it to be a daily driver.
  • crazysim 2 hours ago
    Am I crazy or did the OP swap the contents of the posts around accidentally?

    https://eddie.codes/posts/pandas-should-go-extinct/ <=> https://eddie.codes/posts/source-code-comments/

  • latent-person 56 minutes ago
    In my opinion a better argument to stop using pandas is the very unintuitive API pandas have. Additionally, a slight change in the query can force you to restructure the whole query (change all lines), while in Polars (and tidyverse in R) it's just a simple one-line change.
    • __mharrison__ 20 minutes ago
      If you learn to write pandas correctly you end up writing it very similar to polars (or tidyverse).

      I agree that the API has warts, though typically it is more concise than polars.

      • latent-person 15 minutes ago
        The problem is a small change in the question can force you to make huge changes in the code in pandas. I recently gave some examples in my blog [1]. E.g. compare the last two code blocks, where the small change is just that the median is taken within countries. This requires several line changes in pandas.

        [1]https://bjarkehautop.github.io/Website/blog/data-wrangling-t...

        • anakaine 2 minutes ago
          Personally, I think the inclusion of lambda functions in the pandas code shows that the approach isnt ideal to begin with. Fine on smaller datasets, but typically becomes a slow and cumbersome overhead when interacting with larger datasets. More verbose per line vector operations are generally orders of magnitude faster, and become more so as the data grows.
        • __mharrison__ 2 minutes ago
          I looked at your code very quickly, but it looks like you need to use .filter after a .groupby...
  • aadyachinubhai 24 minutes ago
    The reality is most of the pandas audience don't care about performance. Whenever performance is in the question people have always used polars, duckdb, dask etc. These users are usually software engineers and not data analysts. Agreed, that there is a real gap in latency and performance though!
  • sirfz 14 minutes ago
    I've plugged this several times here (just a fan) but chdb's DataStore is a "lazy" drop-in replacement for pandas dataframe. Pandas API with chdb performance. I don't use it myself (I do everything mostly in sql with either duckdb or chdb) but always found pandas more convinient than polars for quick and dirty data crunching (less typing and frankly more pythonic if you're used to slicing).
  • bijowo1676 1 hour ago
    Strongly disagree with the author.

    For dumb simple select group by OLAP queries on medium data ? Sure clickhouse local or duckdb works perfectly well.

    But if you need to construct dataset ? Or process existing dataset, do heavy filtering, transformation, reshaping, splitting? The proper ETL work, then pandas is really the perfect use case.

    And pandas can work with small memory footprint as well, its actually trivial to do that, plus there are libraries like Modin that are upgrades over pandas with pandas api

    Pandas is the swiss knife tool of data science that lets you do anything with the data and it integrates well with ML libraries

    • minimaxir 1 hour ago
      It's subjective, but IMO polars's APIs are better and than pandas's for ETL. Notably, pandas ported over polars's use of .col() for the 3.0.0 release: https://pandas.pydata.org/docs/whatsnew/v3.0.0.html#initial-...

      Also, pandas's group_by() and window function pipelines made me want to tear my hair out while both are easy in polars.

    • __mharrison__ 19 minutes ago
      Use pandas if you want advanced analytics, visualization, or ml.

      Use SQL if you need to move data around.

      • bijowo1676 11 minutes ago
        How would you use sql to move data from three excel files (some of them may have arbitrary number of worksheets), pack of json files, two csv files, one mysql, one postgres db, some parquets, some of them in S3) ???

        And in the end save processed data in aws s3 in another format like iceberg or whatever

        In pandas a lot of these are one-liners that are impossible in sql (depends on what sql engine you have?)

  • dweinus 58 minutes ago
    Nice post but they quickly disregard Dask, don't explain why, don't test it, even exclude it from the benchmark they quote. I don't know if is the better answer, but it seems worth testing if you want a balance approachable + scalable. That's kind the thing Dask was meant to do.
    • x0xMaximus 39 minutes ago
      We run Dask distributed for contents in what their chart covers (10TB+ with 4+ physical hosts) and it's a problem that can't really be discussed without well defined drive read or network latency limitations (if non-local storage). We've found defining recordsize, raid and other filesystem patterns to have an equal or larger impact in itself. Pandas is great because it allows us to validate on a sampling or smaller time window dataset, and then the same engineers can use a very similar implementation for Dask distributed through their scheduler tool.

      While throwing hardware at problems isn't always a reasonable solution, it's trivialized to discuss this without IOPS or other read limitations.. and the nvme enterprise market is rapidly developing making this change quickly.

  • akdor1154 1 hour ago
    Sup Eddie, the actual motivating example is we finally nixed Pandas from Data Ingestion, now reading excel files takes 2 seconds instead of 2 minutes. However unfortunately your

    > TODO: rewrite this entire service

    remains.

    • __mharrison__ 14 minutes ago
      Rewriting pandas to polars is relatively trivial for most tasks these days. Especially if you wrote your pandas code correctly.

      I still prefer (and use) Pandas for EDA. I think matplotlib integration is a better choice for most viz.

      Also, I'm probably in the top 3-5 worldwide for number of folks I've trained with pandas. I offer Polars training and there is little demand for it.

    • __eddie__ 1 hour ago
      > Sup Eddie, the actual motivating example is we finally nixed Pandas from Data Ingestion, now reading excel files takes 2 seconds instead of 2 minutes. However unfortunately your

      I shall pass this information along to my sleep paralysis demons. They'll be glad to hear it.

      > > TODO: rewrite this entire service

      Part of me enjoys the historical significance of this. However, I feel like keeping it would violate my own princples: https://eddie.codes/posts/source-code-comments/

  • stephenlf 1 hour ago
    Besides the performance benefits, I use Polars at work because it’s just (subjectively) nicer to work with. The “pl.col” API lets you create arbitrary generated/virtual columns anywhere you want, declaratively. You can throw in these column expressions in wherever without actually computing their values and storing that in memory. Very powerful stuff.
  • epihelix 33 minutes ago
    I would love to see a speed comparison with R incorporated into this.
  • db48x 43 minutes ago
    I wonder what they taste like.
  • pjjpo 1 hour ago
    Liked the title
  • willsmith72 2 hours ago
    Makes sense, especially with AI coding tools the rewrite and familiarity arguments hold less water. Similar for the rustify everything crazy.

    The problem is, orgs who see themselves as big data orgs want to act that way, even if they're medium data. "But we'll need it when we grow", "we need to know the state of the art tools"

  • evolve-maz 1 hour ago
    Only in the last few years did I start using SQL properly. Before that my pipelines would live in python. Now I offload as much to the db as possible, and keep my python simple glue. I'm very happy with this compared to other methods in pandas or polars.

    If I still need to do db-like things in python I think duckdb is better.

  • japgolly 1 hour ago
    Nitpick for the author: it looks like you've got (a,b] when you actually mean [a,b). Either that or change the ≥ to be >.
  • jonahss 2 hours ago
    Real link here: https://eddie.codes/posts/source-code-comments/

    Something going wonky on their blog, where two posts got their links swapped.

    • __eddie__ 2 hours ago
      Yep sorry, fixing now. It's what you get for being bitten by the bug to write twice in the same day
    • jivanvl 1 hour ago
      So it wasn’t just then, for me it linked to their “Useful code comments” post
  • dvt 2 hours ago
    I've been saying this since using Databricks at a company almost a decade ago. Most folks do not need big data tools, and it's just so entrenched because everyone wanted to be a "big data" company and pandas was how you handled big data.
  • stephantul 2 hours ago
    Agreed on all counts.

    In many cases I’ve found directly using python primitives to be less confusing than pandas.

    Similarly, in companies I’ve worked at, the datasets just aren’t that big. Especially if you’ve got access to modern hardware.

  • qwertytyyuu 1 hour ago
    Pandas should go extinct? I’m confused Edit: Oh link was broken before
  • Vaslo 2 hours ago
    I use polars or duckdb now exclusively. Better syntax, better performance. But pandas is deeply entrenched - I try to get my team off it but it’s an uphill battle. It’s not going anywhere anytime soon.
  • jgalt212 1 hour ago
    I'd drop pandas if polars worked eamlessly with sklearn.
    • __mharrison__ 12 minutes ago
      Most of the time it works, but there are still corner cases throughout the python ML ecosystem where polars fails.

      This and the decision not to natively plot with matplotlib keep me in pandas for most tasks. (Plus there's is still a relatively large demand for pandas training.)

    • minimaxir 1 hour ago
      You can do to_pandas() and from_pandas() which doesn't have much overhead, or to_numpy() directly which is typically zero-copy and thus no overhead: https://docs.pola.rs/api/python/stable/reference/dataframe/a...
  • ChrisArchitect 2 hours ago
    Title is currently, err...: Useful Code Comments

    Hoping OP can fix this on their end so the url has the expected content. Whoops!

  • viccis 2 hours ago
    Polars seems nice but in my experience using it, the "lazy" APIs would still immediately materialize a ton of stuff in memory and had very spotty support on what data formats and storage integrations were possible with scan_* functions (though that was half a year ago and the support is slowly improving). It's frustrating, I mean really frustrating, to think I could solve a lot of my "scan through heinous amounts of data without any memory hungry things like window aggregations without blowing out my memory" with Polars and then watch my scan_thisorthat() call result in instant memory usage ballooning.

    DuckDB on the other hand is wonderful and truly doesn't use any more memory than it really needs to.

    • __mharrison__ 9 minutes ago
      Did you ever report these issues to polars developers?

      I reported many bugs to both pandas and polars over the years and both teams tend to address relatively big ones. (Still have some outstanding pandas bugs that I think are a big deal but the devs disagree.)

  • jijji 1 hour ago
    so this story is something that's really important for everybody to know about and should not get downvoted...