生成关键字

2019独角兽企业重金招聘Python工程师标准>>>

需要这5个jar

IKAnalyzer2012_u6.jar

lucene-analyzers-3.6.2.jar

lucene-core-3.6.2.jar

lucene-highlighter-3.6.2.jar

lucene-memory-3.6.2.jar

关键字的显示

package com.zte.common.util;

import java.awt.event.ActionEvent;

import java.awt.event.ActionListener;

import java.io.IOException;

import java.io.StringReader;

import java.util.ArrayList;

import java.util.Arrays;

import java.util.Collections;

import java.util.Comparator;

import java.util.HashMap;

import java.util.List;

import java.util.Map;

import java.util.Map.Entry;

import javax.swing.BoxLayout;

import javax.swing.JButton;

import javax.swing.JFrame;

import javax.swing.JPanel;

import javax.swing.JScrollPane;

import javax.swing.JTextArea;

import org.apache.lucene.analysis.TokenStream;

import org.apache.lucene.analysis.tokenattributes.CharTermAttribute;

import org.wltea.analyzer.lucene.IKAnalyzer;

/**

 * 获取文章关键字

 * @author anwj

 *

 */

public class WordUtil {

/** 获取关键字个数 */

private final static Integer NUM = 5;

/** 截取关键字在几个单词以上的数量 */

private final static Integer QUANTITY = 1;

/**

* 传入String类型的文章，智能提取单词放入list中

* @param article

* @param a

* @return

* @throws IOException

*/

private static List<String> extract(String article,Integer a) throws IOException {

List<String> list =new ArrayList<String>();         //定义一个list来接收将要截取出来单词

IKAnalyzer analyzer = new IKAnalyzer();             //初始化IKAnalyzer

analyzer.setUseSmart(true);                         //将IKAnalyzer设置成智能截取

TokenStream tokenStream=                            //调用tokenStream方法(读取文章的字符流)

analyzer.tokenStream("", new StringReader(article));

while (tokenStream.incrementToken()) {              //循环获得截取出来的单词

CharTermAttribute charTermAttribute =           //转换为char类型

tokenStream.getAttribute(CharTermAttribute.class);

String keWord= charTermAttribute.toString();    //转换为String类型

if (keWord.length()>a) {                     //判断截取关键字在几个单词以上的数量(默认为2个单词以上)

list.add(keWord);                           //将最终获得的单词放入list集合中

}

}

return list;

}

/**

* 将list中的集合转换成Map中的key，value为数量默认为1

* @param list

* @return

*/

private static Map<String, Integer> list2Map(List<String> list){

Map<String, Integer> map=new HashMap<String, Integer>();

for(String key:list){                      //循环获得的List集合

if (list.contains(key)) {              //判断这个集合中是否存在该字符串

map.put(key, map.get(key) == null ? 1 : map.get(key)+1);

}                                       //将集中获得的字符串放在map的key键上

}                                          //并计算其value是否有值，如有则+1操作

return map;

}

/**

* 提取关键字方法

* @param article

* @param a

* @param n

* @return

* @throws IOException

*/

public static String[] getKeyWords(String article,Integer a,Integer n) throws IOException {

List<String> keyWordsList= extract(article,a);            //调用提取单词方法

Map<String, Integer> map=list2Map(keyWordsList);      //list转map并计次数

//使用Collections的比较方法进行对map中value的排序

ArrayList<Entry<String, Integer>> list = new ArrayList<Entry<String,Integer>>(map.entrySet());

Collections.sort(list, new Comparator<Map.Entry<String, Integer>>() {

public int compare(Map.Entry<String, Integer> o1, Map.Entry<String, Integer> o2) {

return (o2.getValue() - o1.getValue());

}

});

if (list.size()<n) n=list.size();                    //排序后的长度，以免获得到null的字符

String[] keyWords=new String[n];                    //设置将要输出的关键字数组空间

for(int i=0; i< list.size(); i++) {                  //循环排序后的数组

if (i<n) {                                       //判断个数

keyWords[i]=list.get(i).getKey();           //设置关键字进入数组

}

}

return keyWords;

}

/**

* 

* @param article

* @return

* @throws IOException

*/

public static String[] getKeyWords(String article) throws IOException{

return getKeyWords(article,QUANTITY,NUM);

}

public static void main(String[] args) {

final JTextArea articleTextArea = new JTextArea(40, 20);

JScrollPane articleScroll = new JScrollPane(articleTextArea); 

articleScroll.setHorizontalScrollBarPolicy(JScrollPane.HORIZONTAL_SCROLLBAR_AS_NEEDED); 

articleScroll.setVerticalScrollBarPolicy(JScrollPane.VERTICAL_SCROLLBAR_AS_NEEDED);

final JTextArea keyTextArea = new JTextArea(10, 10);

JScrollPane keyScroll = new JScrollPane(keyTextArea); 

keyScroll.setHorizontalScrollBarPolicy(JScrollPane.HORIZONTAL_SCROLLBAR_AS_NEEDED); 

keyScroll.setVerticalScrollBarPolicy(JScrollPane.VERTICAL_SCROLLBAR_AS_NEEDED);

JButton getKeyBtn = new JButton("生成关键字");

getKeyBtn.addActionListener(new ActionListener() {

@Override

public void actionPerformed(ActionEvent e) {

String articleText = articleTextArea.getText();

try {

String[] keywords = getKeyWords(articleText);

keyTextArea.setText(Arrays.toString(keywords));

} catch (IOException e1) {

e1.printStackTrace();

}

}

});

JPanel mainPanel = new JPanel();

mainPanel.setLayout(new BoxLayout(mainPanel, BoxLayout.Y_AXIS));

mainPanel.add(articleScroll);

mainPanel.add(getKeyBtn);

mainPanel.add(keyScroll);

JFrame frame = new JFrame("提取文章关键字");

frame.add(mainPanel);

frame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);

frame.setSize(600, 300);

frame.setVisible(true);

}

}

转载于:https://my.oschina.net/u/2433838/blog/667919