найти положение элементов в серии 2 серии 1..серия 1: [1,2,3,4,5,6,7] серия 2 [1,3,7] Пример вывода: [0,2,6] - PullRequest
0 голосов
/ 19 февраля 2020

серия 1: [1,2,3,4,5,6,7]
серия 2 [1,3,7]

Пример вывода: [0,2, 6]

import ast,sys
    import pandas as pd
    input_str = sys.stdin.read()
    input_list = ast.literal_eval(input_str)
    series1=pd.Series(input_list[0])
    series2=pd.Series(input_list[1])
    z =series2.size
    c=[]
    for i in range (0,z):
        if series2[i] in series1:
            c.append(series1.index(series2[i]))

out_list=c
print(list(map(int,out_list)))

1 Ответ

0 голосов
/ 19 февраля 2020

Вам нужно использовать серию pandas, или что-то подобное будет работать?

in1 = [1,2,3,4,5,6,7]
in2 = [1,3,7]

output = [in1.index(x) for x in in2]
print(output)

Вывод:

[0, 2, 6]
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...