Java實現詞頻統計

2020-10-15 14:00:46

一、隨便找一篇英文文章,儲存在txt文字中,內容如下:

There are moments in life when you miss someone so much that you just want to pick them from your dreams and hug them for real! Dream what you want to dream;go where you want to go;be what you want to be,because you have only one life and one chance to do all the things you want to do.

  May you have enough happiness to make you sweet,enough trials to make you strong,enough sorrow to keep you human,enough hope to make you happy? Always put yourself in others’shoes.If you feel that it hurts you,it probably hurts the other person, too.

  The happiest of people don’t necessarily have the best of everything;they just make the most of everything that comes along their way.Happiness lies for those who cry,those who hurt, those who have searched,and those who have tried,for only they can appreciate the importance of people

  who have touched their lives.Love begins with a smile,grows with a kiss and ends with a tear.The brightest future will always be based on a forgotten past, you can’t go on well in lifeuntil you let go of your past failures and heartaches.

  When you were born,you were crying and everyone around you was smiling.Live your life so that when you die,you're the one who is smiling and everyone around you is crying.

  Please send this message to those people who mean something to you,to those who have touched your life in one way or another,to those who make you smile when you really need it,to those that make you see the brighter side of things when you are really down,to those who you want to let them know that you appreciate their friendship.And if you don’t, don’t worry,nothing bad will happen to you,you will just miss out on the opportunity to brighten someone’s day with this message.

二、編寫WordCount類實現詞頻統計

package hdfs;

import java.io.BufferedReader;
import java.io.FileReader;
import java.util.Map;
import java.util.TreeMap;
import java.util.Set;
import java.util.TreeSet;
import java.util.Arrays;

/**
 * 功能:統計英文文章的詞頻
 */
public class WordCount {
    /**
     * 功能:傳入路徑字串,返回文字的字串形式
     *
     * @param path
     * @return
     */
    public static String getTxtString(String path) {
        StringBuilder sBuilder = new StringBuilder();
        BufferedReader br = null;
        try {
            br = new BufferedReader(new FileReader(path));
            char[] buffer = new char[512];
            int len;
            while ((len = br.read(buffer)) != -1) {
                sBuilder.append(new String(buffer, 0, len));
            }
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            try {
                if (br != null) br.close();
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
        return sBuilder.toString();
    }

    /**
     * 功能:傳入文字字串,返回單詞詞頻
     *
     * @param txtString
     * @return
     */
    @SuppressWarnings("all")
    public static Map<String, Integer> getCounter(String txtString) {
        Map<String, Integer> treeMap = new TreeMap<>();
        Set<String> treeSet = new TreeSet<>();
        String[] strArr = txtString.split("[\\s?',!;.\u3000]+");
        Arrays.asList(strArr).forEach(treeSet::add);
        for (String s : treeSet) {
            int counter = 0;
            for (String s1 : strArr) {
                if (s.equals(s1)) counter++;
            }
            treeMap.put(s, counter);
        }
        return treeMap;
    }
}

三、建立TestWordCount類進行測試

package hdfs;

import java.util.Map;

public class TestWordCount {
    public static void main(String[] args) {
        String txtString = WordCount.getTxtString("src/main/java/hdfs/article.txt");
        Map<String, Integer> treeMap = WordCount.getCounter(txtString);
        treeMap.forEach((key, value) -> System.out.println(key + ": " + value));
    }
}

執行結果:

Always: 1
And: 1
Dream: 1
Happiness: 1
If: 1
Live: 1
Love: 1
May: 1
Please: 1
The: 2
There: 1
When: 1
a: 4
all: 1
along: 1
always: 1
and: 7
another: 1
appreciate: 2
are: 2
around: 2
bad: 1
based: 1
be: 3
because: 1
begins: 1
best: 1
born: 1
brighten: 1
brighter: 1
brightest: 1
can: 1
can’t: 1
chance: 1
comes: 1
cry: 1
crying: 2
day: 1
die: 1
do: 2
don’t: 3
down: 1
dream: 1
dreams: 1
ends: 1
enough: 4
everyone: 2
everything: 2
failures: 1
feel: 1
for: 3
forgotten: 1
friendship: 1
from: 1
future: 1
go: 4
grows: 1
happen: 1
happiest: 1
happiness: 1
happy: 1
have: 7
heartaches: 1
hope: 1
hug: 1
human: 1
hurt: 1
hurts: 2
if: 1
importance: 1
in: 4
is: 2
it: 3
just: 3
keep: 1
kiss: 1
know: 1
let: 2
lies: 1
life: 4
lifeuntil: 1
lives: 1
make: 6
mean: 1
message: 2
miss: 2
moments: 1
most: 1
much: 1
necessarily: 1
need: 1
nothing: 1
of: 6
on: 3
one: 4
only: 2
opportunity: 1
or: 1
other: 1
others’shoes: 1
out: 1
past: 2
people: 3
person: 1
pick: 1
probably: 1
put: 1
re: 1
real: 1
really: 2
searched: 1
see: 1
send: 1
side: 1
smile: 2
smiling: 2
so: 2
someone: 1
someone’s: 1
something: 1
sorrow: 1
strong: 1
sweet: 1
tear: 1
that: 6
the: 8
their: 3
them: 3
they: 2
things: 2
this: 2
those: 9
to: 19
too: 1
touched: 2
trials: 1
tried: 1
want: 6
was: 1
way: 2
well: 1
were: 2
what: 2
when: 4
where: 1
who: 10
will: 3
with: 4
worry: 1
you: 32
your: 4
yourself: 1