vaexio / vaex

Out-of-Core hybrid Apache Arrow/NumPy DataFrame for Python, ML, visualization and exploration of big tabular data at a billion rows per second 🚀
https://vaex.io
MIT License
8.27k stars 590 forks source link

[FEATURE-REQUEST] take function in Vaex for first n columns #2350

Open ukanchan opened 1 year ago

ukanchan commented 1 year ago

I am using vaex dataframe which has columns > 100. I want to return first n columns from dataframe. Using take function I can return only first n rows but not the columns.

Here is the code: vaexDF.take(range(0,20, 1)) -> return first 20 rows

Pandas has something like pandsDF.take(range(0,20, 1), axis=1) -> returns first 20 columns

When I try to use axis 1 to use vaex take function for columns, it throws exception: take() got an unexpected keyword argument 'axis'

Can vaex provide some argument in take function that will indicates the slicing of rows or columns. ?