Блог

Привет, у меня есть df, как показано ниже:

 Col1 Col2  SM_ SM_ SM_ N_ EX_,SM_ EX_,CO_ SL_,N_ PD_,SL_  

Я хочу сравнить оба столбца и посмотреть, присутствует ли значение в Col1 в Col2 или нет. Несколько значений в обоих столбцах разделены запятыми. Таким образом, в идеале результат сравнения строк, показанного выше, должен быть:

 True False True True  

Прямо сейчас у меня есть следующий код:

 finaldf['C'] = finaldf.apply(lambda x: x.Extracted_pillars in x.min_pillar_score, axis=1)  

Это работает для значений одного кода, но когда каждый столбец содержит несколько значений, это работает неправильно.

Вы можете использовать функцию, которая для каждой ячейки создает набор значений, разделенных запятыми, и возвращает, не является ли пересечение этих наборов в строке пустым:

 import pandas as pd  def columns_overlap(row):  sets = [set(s.split(',')) for s in row]  intersection = set.intersection(*sets)  return intersection != set()  df = pd.DataFrame({'Col1': ['SM_', 'SM_', 'EX_,SM_', 'SL_,N_'],  'Col2': ['SM_', 'N_', 'EX_,CO_', 'PD_,SL_']})  df['C'] = df.apply(columns_overlap, axis=1)  

Это приводит к следующему кадру данных:

 Col1 Col2 C 0 SM_ SM_ True 1 SM_ N_ False 2 EX_,SM_ EX_,CO_ True 3 SL_,N_ PD_,SL_ True  

Ваше решение хорошее, но вам нужно разделить каждую строку запятой и использовать any .

Редактировать:

Оказывается, бывают случаи, когда перед или после запятой стоит пробел. Тогда вы можете использовать:

 finaldf['C'] = finaldf.apply(lambda x: any(i in x.Col1.replace(' ', '').split(',')  for i in x.Col2.replace(' ', '').split(',')),  axis=1)  

Выход:

 0 True 1 False 2 True 3 True