Какие циклы задать для обработки нескольких значений?

Question

Задача в следующем: есть файл s.csv ( во вложении и скриншот ), нужно получить количество раз, где встречается слово VAG и сумму чисел из колонки 'С' по слову VAG. Вообщем все получилось, только я НЕ ПОНИМАЮ, как задать циклы , чтобы автоматом посчитать еще по слову AND и другими словами, если их будет не две а десятки. Прошу показать на моем коде куда вбить циклы. Я полагаю, что должен быть массив String[]str={Vag,And}, который через for each в метод data.contains() прогоняет, но дальше темный лес.

package NewPhoneProj;
import java.io.*;
import java.util.Scanner;
public class CSV {
public static void main(String[] args) {

    String FileName = "c:\\TestJava\\s.csv";

    File file = new File(FileName);

    try {
        Scanner inputStream = new Scanner(file);

        int sum = 0;

        int sumOfnum = 0;

        while (inputStream.hasNext()) {
            String data = inputStream.next();

                if (data.contains("Vag"))
                   {
                String[] values = data.split(";");
                int numbers = Integer.parseInt(values[2]);
                sum += numbers;
                sumOfnum++;



            }
        }
        inputStream.close();
        System.out.println("Sum: "+sum);
        System.out.println("SumOfnum: "+sumOfnum);

    } catch (FileNotFoundException e) {
        e.printStackTrace();

    }

}

}

Answer 1

Вот код который возвращает по каждому уникальному значению первой колонки соответствующую сумму.

public static void main(String[] args) {

    String FileName = "//home//dd//q.csv";

    File file = new File(FileName);

    try {
        Scanner inputStream = new Scanner(file);

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

        while (inputStream.hasNext()) {
            String data = inputStream.next();
            String[] values = data.split(";");
            if (map.get(values[0])==null) {
                map.put(values[0], Integer.parseInt(values[2]));
            } else {
                map.put(values[0], Integer.parseInt(values[2]) + map.get(values[0]));
            }

        }
        inputStream.close();

        for (String key : map.keySet()) {
            System.out.println(key + " " + map.get(key));
        }

    } catch (FileNotFoundException e) {
        e.printStackTrace();

    }

}

Answer 2

Предлагаю дополнительный класс WordInfo с информацией о слове. А также Map<String, WordInfo>, в которой будет она содержаться. В итоге при пробегании в цикле смотрим, если в мапе есть слово - то прибавляем count и добавляем значение в sum, если еще нет - добавляем в мапу с инициализированными значениями.

class WordInfo {
    public int count = 1;
    public int sum;
    public WordInfo(int sum){
        this.sum = sum;
    }
}

main:

public static void main(String[] args) throws java.lang.Exception {
    String FileName = "c:\\TestJava\\s.csv";

    File file = new File(FileName);
    Map<String, WordInfo> wordInfoMap = new HashMap<>();

    try {
        Scanner inputStream = new Scanner(file);

        String[] values; String wordName; Integer wordSum;
        while (inputStream.hasNext()) {
            values = inputStream.next().split(";");
            wordName = values[0];
            wordSum = Integer.parseInt(values[2]);

            if (wordInfoMap.containsKey(wordName)) {
                wordInfoMap.get(wordName).count++;
                wordInfoMap.get(wordName).sum += wordSum;
            } else {
                wordInfoMap.put(wordName, new WordInfo(wordSum));
            }
        }

        inputStream.close();
    } catch (FileNotFoundException e) {
        e.printStackTrace();

    }

    // Итог:
    WordInfo wordInfo;
    for (Map.Entry<String, WordInfo> wordInfoEntry : wordInfoMap.entrySet()) {
        wordInfo = wordInfoEntry.getValue();
        System.out.println("Word \"" + wordInfoEntry.getKey() 
                            + "\". Count: "+ wordInfo.count 
                            + ",  Sum: " + wordInfo.sum);
    }
}