Solved by mariosasko in post #2 
              
              
                Hi, 
this behavior is expected. 
dataset[column] loads the entire dataset column into memory (this is why this part seems so slow to you), so operating directly on this data will be faster than operating on the dataset created with load_dataset, which is memory-mapped, i. e., stored in an arrow file…