File size: 1,280 Bytes
fdffdf0 1c529f8 fdffdf0 1c529f8 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 |
import numpy as np
from scipy.cluster.hierarchy import linkage, fcluster, dendrogram
import matplotlib.pyplot as plt
def passage_outline(matrix,sentences):
Z = linkage(matrix, method="average")
mask = np.eye(matrix.shape[0], dtype=bool)
matrix = np.ma.masked_array(matrix, mask)
median = np.median(matrix)
labels = fcluster(Z, t=median, criterion="distance")
# 根据簇标签和主题句子生成文章结构
structure = {}
for label, sentence in zip(labels, sentences):
if label not in structure:
structure[label] = []
structure[label].append(sentence)
outline = ""
outline_list = []
for key in sorted(structure.keys()):
outline_list.append(f"主题{key}：")
outline = outline+f"主题{key}：\n"
for sentence in structure[key]:
outline_list.append(sentence)
outline = outline+f"- {sentence}\n"
return outline,outline_list
if __name__ == "__main__":
matrix = np.array([[1.0, 0.8, 0.2, 0.1],
[0.8, 1.0, 0.3, 0.2],
[0.2, 0.3, 1.0, 0.9],
[0.1, 0.2, 0.9, 1.0]])
sentences = ["主题句子1", "主题句子2", "主题句子3", "主题句子4"]
print(passage_outline(matrix,sentences)[0])
|