Python 中的高效数据拆分

Question

考虑以下代码

one, two = sales.random_split(0.5, seed=0)
set_1, set_2 = one.random_split(0.5, seed=0)
set_3, set_4 = two.random_split(0.5, seed=0)

我在这段代码中尝试的是将 Sales Sframe（类似于 Pandas DataFrame）中的数据随机分成大约 4 个相等的部分。

什么是 Pythonic/Efficient 实现此目的的方法？

Answer 1

np.random.seed(0)
np.random.shuffle(arr) # in-place
sets = np.array_split(arr, 4)

Efficient splitting of data in Python