Python Pandas Inner объединяет 2 CSV-файла с другим заголовком

Chris K спросил: 13 октября 2017 в 07:31 в: python

Я провел некоторое исследование, но не могу найти ответ для моего дела. У меня есть 2 файла CSV

a.csv

id, fname, lname, address
1, aaa, bbb, ccc
2, abb, ccb, ddd
3, ddd, eee, fff
4, eee, ggg, fff
5, EEE, GGG, RRRR

и 2-й файл

b.csv

ID
1
3
5

желаемый результат будет (заголовок id ниже не является строгим, но id будет в порядке)

id, fname, lname, address
1, aaa, bbb, ccc
3, ddd, eee, fff
5, EEE, GGG, RRRR

что я пробовал

merged = a.merge(b, left_on = ['id'],
                right_on= ['ID'],
                how = 'inner')
merged.to_csv(r'C:\things\output.csv', index=False)

и получил не удается выделить память для ошибки массива ...

edit:

приведенный ниже код будет работать нормально, если заголовки одинаковы (оба идентификаторы), но жизнь не идеальна

merged = a.merge(b, on = 'id')

0 ответов