Use PiecewiseSequence for List take elements #5705

Job	Run time
check-java-publish-build (x86_64-unknown-linux-gnu, runs-on=29842287696/runner=amd64-medium/image...	1m 21s
check-java-publish-build (aarch64-unknown-linux-gnu, runs-on=29842287696/runner=arm64-medium/imag...	1m 57s
Compat check / compat-test	3m 23s
Python (wheel build)	5m 55s
Rust publish dry-run	21s
	12m 57s

Provide feedback