Произвольно выбрать определенные индексы из двух списков, чтобы создать два новых списка

Question

Скажем, у меня есть два списка:

A = ['cat', 'dog', 'cow', 'pig', 'monkey']
B = ['Felix', 'Fido', 'Moo', 'Trump', 'King Kong']

Теперь я хочу создать lists A и list B, выбрав 3 случайных индекса (без замены), сохраняя при этом связь между значениями A и B.

Например,

Случайно выбранные индексы: 4, 0, 3

Следовательно,

A = ['monkey', 'cat', 'pig']
B = ['King Kong', 'Felix', 'Trump']

Есть ли способ длясделать это без написания цикла for, который повторяет 3 раза для выбора случайных индексов?

Answer 1

Вы не можете избежать зацикливания, хотя можете делегировать его функции или использовать в качестве списка.Вы можете зациклить каждый список один раз.

#lets assume your randomly selected numbers are 4, 0 and 3
indices = (4,0,3)
# lets assign an ordinal to each index
indices_ordinals = [(ind, ord) for ind, ord in enumerate(indices)]
# now we can re sort the indices, since we have recorded their original sequence
indices_ordinals.sort(key = lambda x: x[0])
# now go through the lists (for simplicity we assume they have equal number of elemwnts) and fill the output dicts
csi = 0 # csi is the index that we are looking for currently
A1 = {}
B1 = {}
for i, elem in enumerate(A):
    if i == indices_ordinals[csi][0]:
        A1[indices_ordinals[1]] : elem
        csi += 1
csi =0
for i, elem in enumerate(B):
    if i == indices_ordinals[csi][0]:
        B1[indices_ordinals[1]] : elem
        csi += 1
#now we only need to sort A1 and B1 to get desired output
A2=[A1[i] for i in sorted(A1)]
B2=[B1[i] for i in sorted(B1)]

Answer 2

Мне кажется, мне больше нравится почтовый ответ, но я бы выбрал этот подход, используя понимание списка

import random   

l1 = ['cat', 'dog', 'cow', 'pig', 'monkey']
l2 = ['Felix', 'Fido', 'Moo', 'Trump', 'King Kong']
r = [random.randint(0, len(l1)-1) for i in range(3)]

A = [l1[i] for i in r]
B = [l2[i] for i in r]

~/python/stackoverflow/9.23$ python3.7 loop.py 
['monkey', 'cow', 'cat']
['King Kong', 'Moo', 'Felix']

Answer 3

Вы можете zip списков, затем выбрать 3 случайные пары с помощью random.sample и, наконец, снова разделить пары на отдельные списки:

import random

pairs = list(zip(A, B))  # make pairs out of the two lists
pairs = random.sample(pairs, 3)  # pick 3 random pairs
A1, B1 = zip(*pairs)  # separate the pairs

Вот что происходит шаг за шагом:

>>> list(zip(A, B))
[('cat', 'Felix'), ('dog', 'Fido'), ('cow', 'Moo'), ('pig', 'Trump'), ('monkey', 'King Kong')]
>>> random.sample(_, 3)
[('monkey', 'King Kong'), ('pig', 'Trump'), ('dog', 'Fido')]
>>> list(zip(*_))
[('monkey', 'pig', 'dog'), ('King Kong', 'Trump', 'Fido')]

Answer 4

Вы можете использовать:

import numpy as np

indx = np.random.choice(len(A),3,False)

np.array(A)[indx]
Out[593]: array(['cow', 'pig', 'monkey'], dtype='<U6')

np.array(B)[indx]
Out[594]: array(['Moo', 'Trump', 'King Kong'], dtype='<U9')

Answer 5

Python имеет пример функции, которая делает выбор без замены.Вы можете выбрать из индексов и применить выборку к вашим входам.

from random import sample

A = ['cat', 'dog', 'cow', 'pig', 'monkey']
B = ['Felix', 'Fido', 'Moo', 'Trump', 'King Kong']
k = 3

samp = sample(range(len(A)), k)
A_p = [A[i] for i in samp]
B_p = [B[i] for i in samp]