2 ms·

Spark has datasets, which I think lets you do this, although it’s only supported in Java and Scala, not Python.