Сравните каждую строку со всеми строками во фрейме данных и сохраните результаты в списке для каждой строки.

Я пытаюсь сравнить каждую строку со всеми строками в кадре данных pandas с fuzzywuzzy.fuzzy.partial_ratio() >= 85 и записать результаты в список для каждой строки.

Пример:

df = pd.DataFrame({'id': [1, 2, 3, 4, 5, 6], 'name': ['dog', 'cat', 'mad cat', 'good dog', 'bad dog', 'chicken']})

Я хочу использовать функцию pandas с библиотекой fuzzywuzzy, чтобы получить результат:

id  name     match_id_list
1   dog      [4, 5]
2   cat      [3, ]
3   mad cat  [2, ]
4   good dog [1, 5]
5   bad dog  [1, 4]
6   chicken  []

Но я не понимаю, как это получить.


python pandas data-analysis fuzzywuzzy
person pirr    schedule 17.02.2016    source источник

Ответы (1)


arrow_upward
14
arrow_downward

Первым шагом будет поиск индексов, соответствующих условию для данного name. Поскольку partial_ratio принимает только строки, мы apply переносим их в фрейм данных:

name = 'dog'
df.apply(lambda row: (partial_ratio(row['name'], name) >= 85), axis=1)

Затем мы можем использовать enumerate и понимание списка для создания списка индексов true в логическом массиве:

matches = df.apply(lambda row: (partial_ratio(row['name'], name) >= 85), axis=1)
[i for i, x in enumerate(matches) if x]

Поместим все это внутрь функции:

def func(name):
    matches = df.apply(lambda row: (partial_ratio(row['name'], name) >= 85), axis=1)
    return [i for i, x in enumerate(matches) if x]

Теперь мы можем применить функцию ко всему фрейму данных:

df.apply(lambda row: func(row['name']), axis=1)
person IanS    schedule 17.02.2016