cudf.core.column.string.StringMethods.edit_distance_matrix

StringMethods.edit_distance_matrix() → SeriesOrIndex

Computes the edit distance between strings in the series.

The series to compute the matrix should have more than 2 strings and should not contain nulls.

Edit distance is measured based on the Levenshtein edit distance algorithm.

Returns

Series of ListDtype(int64)

Assume N is the length of this series. The return series contains N lists of size N, where the `j`th number in the `i`th row of the series tells the edit distance between the `i`th string and the `j`th string of this series. The matrix is symmetric. Diagonal elements are 0.

Examples

>>> import cudf
>>> s = cudf.Series(['abc', 'bc', 'cba'])
>>> s.str.edit_distance_matrix()
0    [0, 1, 2]
1    [1, 0, 2]
2    [2, 2, 0]
dtype: list