Skip to content

Commit 80f01a4

Browse files
committed
fix variable name bug
1 parent f2a4485 commit 80f01a4

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

dotplot/hierarchical.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
import warnings
2+
23
import pandas as pd
34
from scipy.cluster.hierarchy import linkage, dendrogram, cut_tree
45

@@ -52,7 +53,7 @@ def cluster_hierarchy(data, method, axis, metric='euclidean', n_clusters=None):
5253
_linkage = make_linkage(array, method, metric)
5354

5455
if n_clusters is not None:
55-
cut_result = cut_tree(linkage, n_clusters=n_clusters)
56+
cut_result = cut_tree(_linkage, n_clusters=n_clusters)
5657
df_cut = pd.DataFrame(cut_result.flatten())
5758
label = df_cut.iloc[:, 0].sort_values(ascending=True, inplace=False).index.values
5859
return data.index.values[label]

0 commit comments

Comments
 (0)