That should give you the same elements as your intent, but only
streaming once through the 5 millions records. The order of elements
in the result will be different perhaps, not sure whether it matters.

Perhaps, if the order matters but the result of the first transformation
is rather small , you can post-process it in a second transformation
step doing the sorting.