Pandon Pandas merge, если числа равны

Im пытается объединить два csv на основе состояния. Значение «KEYS» на csv2 должно соответствовать «TCNUM» на CSV1 и добавить третий столбец. Csv очень большие, и это нужно делать с помощью кода.Pandon Pandas merge, если числа равны

df1 - CSV1:

ID          TC_NUM 
dialog_testcase_0101.0001_greeting.xml 101.0001 
dialog_testcase_0101.0002_greeting.xml 101.0002 
dialog_testcase_0101.0003_greeting.xml 101.0003 
dialog_testcase_0101.0004_greeting.xml 101.0004 
dialog_testcase_0101.0005_greeting.xml 101.0005 
dialog_testcase_0101.0006_greeting.xml 101.0006 
dialog_testcase_0901.0008_greeting.xml 901.0007 
dialog_testcase_0101.0008_greeting.xml 101.0008 
dialog_testcase_0501.001_greeting.xml 501.001 
dialog_testcase_0801.0011_greeting.xml 801.0011

df2 - CSV2:

KEYS    TC_NUM 
FIT-3982 TC 101.0011, 101.0004 
FIT-3980 TC 801.0011.901.007 
FIT-3979 TC 101.0006, 501.001, 1907.0019, 1907.0020, 1907.0021

Что я хочу:

csvFinal:

ID          TC_NUM  Keys 
dialog_testcase_0101.0001_greeting.xml 101.0011  FIT-3982 
dialog_testcase_0101.0002_greeting.xml 101.0002  
dialog_testcase_0101.0003_greeting.xml 101.0006  FIT_3979 
dialog_testcase_0101.0004_greeting.xml 101.0004  FIT-3982 
dialog_testcase_0101.0005_greeting.xml 101.0005 
dialog_testcase_0101.0006_greeting.xml 101.0011  FIT_3982 
dialog_testcase_0901.0008_greeting.xml 901.0007  FIT_3979 
dialog_testcase_0101.0008_greeting.xml 101.0008 
dialog_testcase_0501.001_greeting.xml 501.001  FIT-3979 
dialog_testcase_0801.0011_greeting.xml 801.0011  FIT-3980

Мой код ..

mergedOpen = pd.merge(df1, df2, on=['TC_NUM']) 
mergedOpen.set_index('TC_NUM', inplace=True) 

mergedOpen.to_csv('MergedCSVOPEN.csv')

источник

2016-05-18 Anekdotin

Вы можете после set_index удалить первый символ 3 из колонки TC_NUM, split по , с unstack и reset_index создать новый DataFrame для merge. Обе колонки TC_NUM должны быть установлены равными dtype - string или numeric. Я выбираю numeric, поэтому я конвертировать столбец df2.TC_NUMto_numeric:

df2.set_index('KEYS',inplace=True) 

df2 = df2.TC_NUM.str[3:] 
       .str.split(', ', expand=True) 
       .unstack() 
       .reset_index(drop=True, level=0) 
       .reset_index(name='TC_NUM') 

df2['TC_NUM'] = pd.to_numeric(df2['TC_NUM']) 
print (df2) 
     KEYS  TC_NUM 
0 FIT-3982 101.0011 
1 FIT-3980 801.0011 
2 FIT-3979 101.0006 
3 FIT-3982 101.0004 
4 FIT-3980 901.0070 
5 FIT-3979 501.0010 
6 FIT-3982  NaN 
7 FIT-3980  NaN 
8 FIT-3979 1907.0019 
9 FIT-3982  NaN 
10 FIT-3980  NaN 
11 FIT-3979 1907.0020 
12 FIT-3982  NaN 
13 FIT-3980  NaN 
14 FIT-3979 1907.0021

mergedOpen = pd.merge(df1, df2, on='TC_NUM', how='left') 
print (mergedOpen) 
             ID TC_NUM  KEYS 
0 dialog_testcase_0101.0001_greeting.xml 101.0001  NaN 
1 dialog_testcase_0101.0002_greeting.xml 101.0002  NaN 
2 dialog_testcase_0101.0003_greeting.xml 101.0003  NaN 
3 dialog_testcase_0101.0004_greeting.xml 101.0004 FIT-3982 
4 dialog_testcase_0101.0005_greeting.xml 101.0005  NaN 
5 dialog_testcase_0101.0006_greeting.xml 101.0006 FIT-3979 
6 dialog_testcase_0901.0008_greeting.xml 901.0007  NaN 
7 dialog_testcase_0101.0008_greeting.xml 101.0008  NaN 
8 dialog_testcase_0501.001_greeting.xml 501.0010 FIT-3979 
9 dialog_testcase_0801.0011_greeting.xml 801.0011 FIT-3980 

mergedOpen.set_index('TC_NUM', inplace=True) 
mergedOpen.to_csv('MergedCSVOPEN.csv')

источник

2016-05-18 21:23:41 jezrael

Вы ответили на мой последний сложный вопрос, как хорошо. Спасибо, мужик. Попытка это – Anekdotin

Я собираюсь принять это как концепцию основного права для моего скрипта. Я собираюсь спросить следующий шаг, как только я получу эту работу, хотя :) Надеюсь, что вокруг – Anekdotin

ValueError: невозможно разобрать строку – Anekdotin

Pandon Pandas merge, если числа равны

ответ

Смежные вопросы