Как мне объединить_first, индексируя конкретные два столбца DataFrames в один?

После сопоставления по столбцу c, если столбец A имеет значение, вставьте значение столбца A; если нет, вставьте столбец B.

data1                               data2

a    b    c                      a    c    d
a1   b1   c1                     1a   c1   1d   
     b2   c2                     2a   c2   2d
a3        c3                     3a   c3   3d
                                 4a   c4   4d

Результат, который я хочу

  result
   a    b   c     
   a1   b1  c1
   2a   b2  c2
   a3       c3

Я попробовал следующее, но я не был удовлетворен.

->>> result = data1.merge(data2, on=['c'])
Prefixes _x and _y are created. combine_first is not applied.

->>> result = data1.combine_first(data2)
It is not mapped by column c.

Как мне получить хорошие результаты? Я прошу вашей помощи. благодарю вас

3 ответа

Решение

Использование настройки @IdoS:

import pandas as pd
data1 = pd.DataFrame({'a': ['a1', None, 'a3'],
                      'b': ['b1', 'b2', None],
                      'c': ['c1', 'c2', 'c3']})

data2 = pd.DataFrame({'a': ['1a', '2a', '3a', '4a'],
                      'c': ['c1', 'c2', 'c3', 'c4'],
                      'd': ['1d', '2d', '3d', '4d']})

Ты можешь использовать set_index, combine_firstи переиндексировать:

df_out = data1.set_index('c').combine_first(data2.set_index('c'))\
     .reindex(data1.c)\
     .reset_index()

df_out

Выход:

    c   a     b   d
0  c1  a1    b1  1d
1  c2  2a    b2  2d
2  c3  a3  None  3d

Вы также можете попробовать этот способ:

# set indexes
data1 = data1.set_index('c')
data2 = data2.set_index('c')

# join data on indexes
datax = data1.join(data2.drop('d', axis=1), rsuffix='_rr').reset_index()

# fill missing value in column a
datax['a'] = datax['a'].fillna(datax['a_rr'])

# drop unwanted columns
datax.drop('a_rr', axis=1, inplace=True)

# fill missing values with blank spaces
datax.fillna('', inplace=True)

# output
    a   b   c
0   a1  b1  c1
1   2a  b2  c2
2   a3      c3

# data used
data1 = pd.DataFrame({'a':['a1','','a3'],
                      'b':['b1','b2',''],
                      'c':['c1','c2','c3']})

data2 = pd.DataFrame({'a':['1a','2a','3a','4a'],
                      'c':['c1','c2','c3','c4'],
                      'd':['1d','2d','3d','4d']})

Я не на 100% понимаю, как вы проиндексировали свои фреймы данных (data1 а также data2), но если индексировать их по столбцу 'c' он должен работать.

Вот как я создал ваши данные:

import pandas as pd
data1 = pd.DataFrame({'a': ['a1', None, 'a3'],
                      'b': ['b1', 'b2', None],
                      'c': ['c1', 'c2', 'c3']})

data2 = pd.DataFrame({'a': ['1a', '2a', '3a', '4a'],
                      'c': ['c1', 'c2', 'c3', 'c4'],
                      'd': ['1d', '2d', '3d', '4d']})

Затем я устанавливаю индекс обоих столбцов 'c':

data1 = data1.set_index('c')
data2 = data2.set_index('c')

Тогда я использую combine_first как ты делаешь:

data_combined = data1.combine_first(data_2)

И я получаю это:

    a   b   d
c           
c1  a1  b1  1d
c2  2a  b2  2d
c3  a3  None    3d
c4  4a  NaN 4d

Не уверен, почему вы не хотите строки с индексом 'c4' или столбец 'd', но их достаточно легко удалить:

data_combined = data_combined.drop('d', axis=1)
data_combined = data_combined.loc[data_combined.index != 'c4']

И затем я делаю некоторый повторный заказ, чтобы получить желаемый результат:

data_combined = data_combined.reset_index()
data_combined = data_combined[['a', 'b', 'c']]
data_combined = data_combined.fillna('')


    a   b   c
0   a1  b1  c1
1   2a  b2  c2
2   a3      c3
Другие вопросы по тегам