Найдите разницу между двумя pandas кадрами и добавьте все в новый DF - PullRequest
Новая
       8

Найдите разницу между двумя pandas кадрами и добавьте все в новый DF

1 голос
/ 20 января 2020

У меня есть два Guest-Lists в DataFrames, и я хочу знать, какие имена выпали, какие были добавлены, а затем получить новый DF, что мне делать?

Два DF: 

list1 = {'First Name': ['Adi', 'Beni', 'Dimi', 'Sergei'], 'Last Name': ['Beer', 'Wine', 'Liquor', 'Vodka'], 'Job': ['Brewer', 'Farmer', 'Shopowner', 'Guest']}

list2 = {'First Name': ['Adi', 'Beni', 'Sergei', 'Don'], 'Last Name': ['Beer', 'Wine', 'Vodka', 'Brown']}

Если `` NameX находится в list1 и list2, создайте новый DataFrame с именами в обоих.

Если NameX был в list1, но выпал в list2, покажите мне в новом df.

И если NameX только в list2, покажите мне это в другом дф.

Как мне это сделать?

Между прочим, это преднамеренно, что запись «Работа» только в одном df. Это цель, чтобы сохранить его в новых DFS.

Ответы [ 2 ]

1 голос
/ 20 января 2020

pandas.merge(...) - это то, что вы ищете: https://pandas.pydata.org/pandas-docs/stable/reference/api/pandas.DataFrame.merge.html

Чтобы получить все параметры одновременно, просто выполните: 

df1=pd.DataFrame(list1)
df2=pd.DataFrame(list2)

df3=pd.merge(df1, df2, on=["First Name", "Last Name"], how="outer", indicator=True)

Вывод: 

  First Name Last Name        Job      _merge
0        Adi      Beer     Brewer        both
1       Beni      Wine     Farmer        both
2       Dimi    Liquor  Shopowner   left_only
3     Sergei     Vodka      Guest        both
4        Don     Brown        NaN  right_only
1 голос
/ 20 января 2020

Самый простой способ - соединить имена и фамилии, сравнить их, а затем снова разделить.

Вот соответствующий код: 

FIRST_NAME = 'First Name'
LAST_NAME = 'Last Name'
JOB = 'Job'
KEYS = [FIRST_NAME, LAST_NAME, JOB]


def intersection(lst1, lst2):
    return [value for value in lst1 if value in lst2]


def unzip(lst):
    a = []
    b = []
    for x, y in lst:
        a.append(x)
        b.append(y)
    return a, b


def get_jobs(dct, lst):
    return [dct.get(x) for x in lst]


def get_dict(lst, dict0, keys=KEYS):
    a, b = unzip(lst)
    values = [a, b, get_jobs(dict0, lst)]
    return dict(zip(keys, values))


list1 = {FIRST_NAME: ['Adi', 'Beni', 'Dimi', 'Sergei'], LAST_NAME: ['Beer', 'Wine', 'Liquor', 'Vodka'],
         JOB: ['Brewer', 'Farmer', 'Shopowner', 'Guest']}
list2 = {FIRST_NAME: ['Adi', 'Beni', 'Sergei', 'Don'], LAST_NAME: ['Beer', 'Wine', 'Vodka', 'Brown']}

mapped1 = list(zip(list1.get(FIRST_NAME), list1.get(LAST_NAME)))
mapped2 = list(zip(list2.get(FIRST_NAME), list2.get(LAST_NAME)))
dict_jobs = dict(zip(mapped1, list1.get(JOB)))

intersec = intersection(mapped1, mapped2)
left = []
right = []

for element in set(mapped1) ^ set(mapped2):
    left.append(element) if element in mapped1 else right.append(element)

dict_intersec = get_dict(intersec, dict_jobs)
dict_left = get_dict(left, dict_jobs)
dict_rigth = get_dict(right, dict_jobs)

Вывод: 

{'First Name': ['Adi', 'Beni', 'Sergei'], 'Last Name': ['Beer', 'Wine', 'Vodka'], 'Job': ['Brewer', 'Farmer', 'Guest']}
{'First Name': ['Dimi'], 'Last Name': ['Liquor'], 'Job': ['Shopowner']}
{'First Name': ['Don'], 'Last Name': ['Brown'], 'Job': [None]}

В dict_jobs объединенные имена сопоставляются с их заданием. intersec - список, содержащий имена, содержащиеся в обоих исходных списках. left содержит имена, которые появляются только в первом списке, right только имена, которые появляются во втором списке.

...