В Python, как мне объединить два массива по ключевому столбцу?

Question

Предположим, у меня есть два массива (после импорта numpy как np):

a=np.array([['a',1],['b',2]],dtype=object)

и

b=np.array([['b',3],['c',4]],dtype=object)

Как получить:

c=np.array([['a',1,None],['b',2,3],['c',None,4]],dtype=object)

В основном, объединение с использованием первого столбца в качестве ключа.

Спасибо

Answer 1

Чистый подход Python для этого был бы

da = dict(a)
db = dict(b)
c = np.array([(k, da.get(k), db.get(k))
              for k in set(da.iterkeys()).union(db.iterkeys())])

Но если вы используете NumPy, ваши массивы, вероятно, большие, и вы ищете решение с лучшей производительностью.В этом случае я предлагаю использовать для этого некоторую реальную базу данных, например модуль sqlite3 , который поставляется с Python.

Answer 2

Лучшее решение, которое я нашел, - это использование панд, которые очень хорошо справляются с соединениями, а объекты панд легко преобразуются в / из массивов.