代码难题：如何用Java处理CSV文件，并进行分组统计？

原创傷城~ 2024-12-05 06:36 3阅读 0赞

在Java中处理CSV文件并进行分组统计，你可以使用开源库如Apache Commons CSV和GROUPING功能。

以下是一个基本的步骤：

1. 导入所需的库：
```java
import org.apache.commons.csv.CSVReader;
import org.apache.commons.collections4.MapTransformer;
import org.apache.commons.collections4.grouping.GroupingStrategy;
```

2. 读取CSV文件：
```java
CSVReader reader = new CSVReader(new FileReader("your_file.csv")));

List<String[]> rows = reader.readAll();
reader.close();
```

3. 分组统计：
假设你想要按照某一列（例如：group_column）进行分组，然后对每个分组计算特定列（例如：value_column）的总和。

首先，创建一个分组策略类：

```java
public class GroupByColumnStrategy implements GroupingStrategy<String[]> {

private final String groupColumn;
    private final String valueColumn;

public GroupByColumnStrategy(String groupColumn, String valueColumn) {
        this.groupColumn = groupColumn;
        this.valueColumn = valueColumn;
    }

@Override
    public Collection<Grouped<String[], Object>>> group(List<String[]>> rows) {
        Map<String, Long>> groupedValuesMap = new HashMap<>();

for (String[] row : rows) {
            String groupName = row[groupColumn];
            Long currentValue = groupedValuesMap.get(groupName);
            if (currentValue == null) {
                currentValue = 0L;
            }
            currentValue += Long.parseLong(row[valueColumn]));
            groupedValuesMap.put(groupName, currentValue);
        }

return groupedValuesMap.values().stream()
                .collect(Collectors.toList()));
    }
}
```

4. 在主程序中使用这个分组策略进行统计：

```java
// 示例：假设group_column为"country", value_column为"sales"
GroupByColumnStrategy strategy = new GroupByColumnStrategy("country", "sales"));

List<Object> result = strategy.group(rows).stream()
        .map(groupValue -> groupValue.getValue() * 1.0 // 如果你的值是数字，这里可以乘以一个系数
                        + groupValue.getKey() // 加上分组键作为描述信息
                ))
        .collect(Collectors.toList());

// 结果是一个包含销售额和分组键的列表

```

这个示例假设你有一个CSV文件，每行包含"country", "sales"两列数据。代码会根据指定的group_column和value_column对数据进行分组统计，并计算每个分组的销售额总和。