JavaでCSVを出力する際にカテゴリー別に小計行を追加したい。

Question

加工元データ

|NO|CATEGORY|PRICE|
|:--|:--|：--|
|1|Apple|100|
|2|Apple|100|
|3|Apple|100|
|4|Banana|100|
|5|Banana|100|
|6|Banana|100|
|7|Banana|100|
|8|Peach|100|
|9|Peach|100|
|10|Peach|100|

加工後データ（CSV）

|NO|CATEGORY|PRICE|
|:--|:--|：--|
|1|Apple|100|
|2|Apple|100|
|3|Apple|100|
||小計|300
|4|Banana|100|
|5|Banana|100|
|6|Banana|100|
|7|Banana|100|
||小計|400|
|8|Peach|100|
|9|Peach|100|
|10|Peach|100|
||小計|300|

したいことの説明

【加工元データ】のカテゴリー別の小計を計算するループ処理で、
各カテゴリーの下に小計の行を追加したいです。　＊【加工後データ（CSV）参照】

@Getter
@Setter
class SumTable{
  private Integer no;
  private String category;
  private Integer price;
}

// 加工元データリスト
List<SumTable> list = 加工元データをリストで取得;

// 加工後データ格納リスト
List<SumTable> csvList = new List<SumTable>();

// 集計処理
Map<String,SumTable> map = new LinkedHashMap<>();
for(SumTable s : list){
  if(map.containsKey(s.getCategory())){
    
  } else {
    
  }
}

// CSV出力
return convertCsv(csvList);

小計行について

・No列の値は空文字
・Category列はCategory
・Price列はCategory別の合計

理想

本当に聞きたいことは、カテゴリーの列は複数あって
カテゴリー１＞カテゴリー２＞カテゴリー３ごとの集計をとる必要があります。
その点の背景を考慮していただきご教授していただけると非常に助かります。

Comments

[dodox86]

CSVの元データで、カテゴリーが必ず連続しているのであればカテゴリーが変わった時点で集計結果を1行出力すればよいだけだし、カテゴリーがとびとびに登録されているのであれば、集計後に並べ直して出力する必要があります。

[BomBooWood]

カテゴリーが変わった時点のコードの書き方に戸惑っている状況です。

Answer 1

1. キーになる項目ごとに分ける。
2. 分けたものを出力する
3. 分けたものを集計して出力する
4. キーが複数あるなら、分けたものをさらに分ける

Javaで書くのはめんどうなので、サンプルはKotlinで書きます。StreamにもgroupBy(), sum() があるはずなので書けるでしょう。もっともStreamを使わずとも書けます。

object HogeSum {
  @JvmStatic
  fun main(args: Array<String>) {
    dataList
      .groupBy { data: Data -> data.catagory }
      .forEach { (catagory: String, list: List<Data>) ->
        list.forEach { data: Data ->
          println(data.toCSV())
        }

        val sum = list.sumOf { data: Data -> data.price }
        println(listOf("", "小径(${catagory})", sum).joinToString(","))
      }
  }

  //
  data class Data(
    val no: Int,
    val catagory: String,
    val price: Int
  ) {
    fun toCSV() =
      listOf(no, catagory, price).joinToString(",")
  }

  //
  val dataList = listOf(
    Data(1, "Apple", 100),
    Data(2, "Apple", 100),
    Data(3, "Apple", 100),
    Data(4, "Banana", 100),
    Data(5, "Banana", 100),
    Data(6, "Banana", 100),
    Data(7, "Banana", 100),
    Data(8, "Peach", 100),
    Data(9, "Peach", 100),
    Data(10, "Peach", 100)
  )
    .shuffled() // あえて、まぜる
}

元データをかき回しているので順序はランダムです。

10,Peach,100
9,Peach,100
8,Peach,100
,小径(Peach),300
1,Apple,100
2,Apple,100
3,Apple,100
,小径(Apple),300
4,Banana,100
7,Banana,100
6,Banana,100
5,Banana,100
,小径(Banana),400