在 Python 中拆分另一个数组

Question

我正在尝试编写一个代码来比较两个数组（'a' 和 'b)' 并做类似的事情，得到 'c' 数组：

a = [1, 2, 2, 3, 3, 5]

b = [1, 2, 3] (随机取自'a')

c = [2, 3, 5]

我的问题很简单，算法没问题。我使用的代码是：

vetor1 = [1, 2, 2, 3, 3, 5]
vetor2 = sorted(random.sample(vetor1, 3))
inter = np.intersect1d(vetor1, vetor2)

cont_array2 = []
for i in range(len(inter)):
    cont2 = 0
    for j in range(len(vetor2)):
        if inter[i]==vetor2[j]:
            cont2 = cont2 + 1   
    cont_array2.append(cont2)

cont_array1 = []
for i in range(0,len(inter),1):
    cont1 = 0
    for j in range(0, len(vetor1), 1):
        if inter[i]==vetor1[j]:
            cont1 = cont1 + 1   
    cont_array1.append(cont1)

n_elements = np.subtract(cont_array1, cont_array2)

vetor2_1 = []
for i in range(len(n_elements)):
    if n_elements[i]!=0:
        vetor2_1.append(inter[i]*np.ones(n_elements[i]))

vetor2_2 = []
for data1 in vetor1:
    if data1 not in inter:
        vetor2_2.append(data1)

vetor3 = sorted(vetor2_1 + vetor2_2)
print(vetor3)

我得到的结果是：

[array([2.]), array([3.]), 5]

我想问题出在：

vetor2_1.append(inter[i]*np.ones(n_elements[i]))

因为：

print(vetor2_1)

给我：

[array([2.]), array([3.])]

我没有收到答案 [2, 3, 5]，我该怎么办？

Answer 1

IIUC，你可以把你所有的算法都写成一行：

a = [1, 2, 2, 3, 3, 5]

b = [1, 2, 3]

c = [a.pop(a.index(i)) for i in b] #Works even with strings.

输出：

[2, 3, 5]

如果你需要参数化一些东西，你可以把它变成一个函数。这将适用于 sorted(random.sample(vetor1, 3))，但我没有将其用于可复制性。

Answer 2

//generating a random index for array a, to take 3 elements out of it
my_random_int = random.randint(0,len(a)-2)

//popping the element out of the array to put them in b

for x in range(0,2):
    b[x]=a.pop(my_random_int+x)

来源：

https://www.programiz.com/python-programming/methods/list/pop

https://www.edureka.co/blog/generate-random-number-in-python/

在 Python 中拆分另一个数组

Split an array inside another in Python

python

arrays

numpy

list

numpy-slicing

输出：