Как объединить 2 массива pandas по общему столбцу?

Пользователь

от ransom_homenick , в категории: Python , 7 месяцев назад

Как объединить 2 массива pandas по общему столбцу?

Facebook Vk Ok Twitter LinkedIn Telegram Whatsapp

1 ответ

Пользователь

от jorge , 6 месяцев назад

@ransom_homenick 

Для объединения двух массивов pandas по общему столбцу можно использовать метод merge(). Например, предположим, что у нас есть два массива df1 и df2, и мы хотим объединить их по столбцу id.


Пример кода:

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
import pandas as pd

# Создание первого массива
df1 = pd.DataFrame({'id': [1, 2, 3],
                   'name': ['John', 'Jane', 'Steve']})

# Создание второго массива
df2 = pd.DataFrame({'id': [2, 3, 4],
                   'age': [25, 30, 35]})

# Объединение массивов по столбцу 'id'
merged_df = pd.merge(df1, df2, on='id')

print(merged_df)


Результат:

1
2
3
   id  name  age
0   2  Jane   25
1   3  Steve   30


В результирующем массиве merged_df останутся только строки, в которых значения столбца id совпадают в обоих массивах.