使用Python进行Jaccard距离的非对称计算

时间:2019-04-03 05:22:28

标签: python distance unsupervised-learning asymmetric

我有一些SAS编码,试图将其转换为Python。我在计算非对称数据上的jaccard距离时遇到了困难-在计算中应忽略零。我确实在jaccard上找到了一些示例,但它们不计算不对称距离。在尝试重新发明轮子之前,只需检查一下图书馆是否有可用的图书馆。如果有人可以引导我朝正确的方向发展,我将不胜感激。

我的测试数据集包含5个标题和5行

H0  H1  H2  H3  H4

A  1  1  1  1  0

B  1  0  1  1  0

C  1  1  1  1  0

D  0  0  1  1  1

E  1  1  0  1  0

以下是通过速记以及使用SAS计算得出的预期结果(距离):

. |  A   |    B   |    C   |   D   |   E

A |  0   |    0.25|    0   |   0.6 |   0.25

B |  0.25|    0   |    0.25|   0.5 |   0.5

C |  0   |    0.25|    0   |   0.6 |   0.25

D |  0.6 |    0.5 |    0.6 |   0   |   0.8

E |  0.25|    0.5 |    0.25|   0.8 |   0        

但是,在python中使用jaccard,我得到的结果如下:

.  |A    |   B   |   C   |   D  |   E

A  |1.00 | 0.43  |  0.61 | 0.55 |   0.46

B  |0.43 | 1.00  |  0.52 | 0.56 |   0.49

C  |0.61 | 0.52  |  1.00 | 0.48 |   0.53

D  |0.55 | 0.56  |  0.48 | 1.00 |   0.49

E  |0.46 | 0.49  |  0.53 | 0.49 |   1.00

下面是我尝试过的代码。我是Python的新手,所以我可能会犯一个明显的错误。我在底部添加了SAS代码,以防有人参考:

Python代码:

np.random.seed(0)
df = pd.DataFrame(np.random.binomial(1, 0.5, size=(100, 5)), 
columns=list('ABCDE'))
print(df.head())

jac_sim = 1 - pairwise_distances(df.T, metric = "jaccard")
jac_sim = pd.DataFrame(jac_sim, index=df.columns, columns=df.columns)

import itertools
sim_df = pd.DataFrame(np.ones((5, 5)), index=df.columns, columns=df.columns)
for col_pair in itertools.combinations(df.columns, 2):
    sim_df.loc[col_pair] = sim_df.loc[tuple(reversed(col_pair))] = 
    jaccard_similarity_score(df[col_pair[0]], df[col_pair[1]])
print(sim_df)

SAS代码:

proc import datafile = '/home/xxx/xxx.csv'  
 out = work.Binary2 replace
 dbms = CSV;
 GUESSINGROWS=MAX;
run;
proc sort;
by VAR1;
run;
title ’Data Clustering of BN’;
proc distance data=Binary2 method=djaccard absent=0 out=distjacc;
var anominal (r0--r4);
id VAR1;
run;

1 个答案:

答案 0 :(得分:1)

我发现了一些明显的错误。首先,您需要创建size=(5,5)的矩阵:

import pandas as pd
import numpy as np
from sklearn.metrics import pairwise_distances, jaccard_similarity_score

np.random.seed(0)
df = pd.DataFrame(np.random.binomial(1, 0.5, size=(5, 5)).T, columns=list('ABCDE'))
print(df.T)

第二件事是,如果仅打印正面,则看不到矩阵有多于5行。仅用5行,这两行:

print(df.T.head())

print(df.T)

打印相同的结果:

   0  1  2  3  4
A  1  1  1  1  0
B  1  0  1  1  0
C  1  1  1  1  0
D  0  0  1  1  1
E  1  1  0  1  0

完成上述更改后,可以使用pairwise_distances

jac_sim = pairwise_distances(df.T.astype(bool), metric = "jaccard")
jac_sim = pd.DataFrame(jac_sim, index=df.columns, columns=df.columns)
print(jac_sim)

为了获得所需的结果:

      A     B     C    D     E
A  0.00  0.25  0.00  0.6  0.25
B  0.25  0.00  0.25  0.5  0.50
C  0.00  0.25  0.00  0.6  0.25
D  0.60  0.50  0.60  0.0  0.80
E  0.25  0.50  0.25  0.8  0.00

上面的代码中还包含.astype(bool),以防止在运行pairwise_distance时出现警告。

在应用转置.T时必须要小心,因为pairwise_distance似乎适用于列而不是行。

具有功能jaccard_similarity_score

import itertools
sim_df = pd.DataFrame(np.zeros((5, 5)), index=df.columns, columns=df.columns)
for col_pair in itertools.combinations(df.columns, 2):
    sim_df.loc[col_pair] = sim_df.loc[tuple(reversed(col_pair))] = \
        1 - jaccard_similarity_score(df[col_pair[0]], df[col_pair[1]], normalize = True)
print(sim_df)

我得到了另一个矩阵:

     A    B    C    D    E
A  0.0  0.2  0.0  0.6  0.2
B  0.2  0.0  0.2  0.4  0.4
C  0.0  0.2  0.0  0.6  0.2
D  0.6  0.4  0.6  0.0  0.8
E  0.2  0.4  0.2  0.8  0.0

仔细观察jaccard_similarity_score

print(df['A'])
print(df['B'])
jaccard_similarity_score(df['A'], df['B'], normalize = True)

显示未排除零的结果:

0    1
1    1
2    1
3    1
4    0
Name: A, dtype: int32
0    1
1    0
2    1
3    1
4    0
Name: B, dtype: int32
Out[123]: 0.8

因为结果是4个相似/ 5个总数= 0.8,而不是3个相似非零/ 4个总非零= 0.75。