python - 根据百分比水平拆分 2D numpy 数组

Question

我希望能够将 2D numpy 水平 分成两部分(80% 和 20%)。我试过使用 np.vsplit() 但它似乎不是为这种情况而设计的。例如，假设我有以下大小矩阵 (6,3)。我想将它水平拆分为 80% 和 20% [大致为 (5,3), (1,3)]，所以我尝试了这样的操作:

M = [[1,2,3],[4,5,6],[7,8,9], [10,11,12], [77,54,11], [424,78,98]]
M = np.asarray(M)
arr1 = np.vsplit(M, int(M.shape[0]* 0.8))[0]  # 80% of data goes to arr1
arr2 = np.vsplit(M, int(M.shape[0]* 0.2))[1]  # 20% of data goes to arr2

我知道这个尝试是不正确的，但我无法修复它(实际上还在学习 python)。如果有人可以帮助修改此代码，请提供帮助。谢谢

Answer 1

您可以使用索引(或使用 train_test_split )这样做:

M = [[1,2,3],[4,5,6],[7,8,9], [10,11,12], [77,54,11], [424,78,98]]
M = np.asarray(M)

split_horizontally_idx = int(M.shape[0]* 0.8) # integer for line selection (horizontal selection)

array1 = M[:split_horizontally_idx , :] # indexing/selection of the 80%
array2 = M[split_horizontally_idx: , :] # indexing/selection of the remaining 20%