8

各行を 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  []

しかし、これを取得する方法がわかりません。

4

1 に答える 1

15

最初のステップは、特定の の条件に一致するインデックスを見つけること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)
于 2016-02-17T15:27:14.673 に答える