如何根据百分比将列表分为两部分？

Question

我需要将阵列分成两部分，第一部分需要有前90%的空间，下一部分应该有剩余空间。但是我只得到了第二个数组的修正结果

例：从11500开始，我得到1150的测试图像，但我得到11500的测试图像

tain_images, test_images = np.split(imagesArr, [int(len(imagesArr)*0.9)])

Answer 1

你的代码对我有用。这可能是因为您在某个地方有一个输入错误，例如代码后面某个地方的tain_images中有一个输入错误

Answer 2

这对我很有用：

import numpy as np
imagesArr = []
for i in range(11500):
    imagesArr.append(str(i+1))
imagesArr = np.array(imagesArr)
tain_images, test_images = np.split(imagesArr, [int(len(imagesArr)*0.9)])
print(len(tain_images))
print(len(test_images))

结果:

10350
1150

Answer 3

您应该使用sklearn^{}

from sklearn.model_selection import train_test_split

train_images, test_images  = train_test_split(imagesArr, test_size=0.1)    

根据所需的测试数据集百分比更改test_size值