Java POI Excel 折线图生成教程 - 使用Apache POI库绘制数据折线图

Java POI Excel 折线图生成教程

本教程将指导您使用Apache POI库在Java程序中生成Excel折线图，并保存为图片文件。示例代码展示了如何从Excel数据创建折线图，设置图例、标题和轴，并最终将图表绘制到图片。

1. 项目准备

首先，您需要在项目中添加 Apache POI 库的依赖。您可以使用 Maven 或 Gradle 等构建工具添加依赖。

Maven 依赖：

<dependency>
    <groupId>org.apache.poi</groupId>
    <artifactId>poi-ooxml</artifactId>
    <version>5.2.3</version>
</dependency>

Gradle 依赖：

implementation 'org.apache.poi:poi-ooxml:5.2.3'

2. 代码示例

import org.apache.poi.ss.usermodel.*;
import org.apache.poi.ss.util.CellRangeAddress;
import org.apache.poi.xddf.usermodel.chart.*;
import org.apache.poi.xssf.usermodel.*;

import javax.imageio.ImageIO;
import java.awt.*;
import java.awt.image.BufferedImage;
import java.io.*;

public class ExcelChartExample02 {

    public static void main(String[] args) {
        try {
            // 读取Excel文件
            FileInputStream fileIn = new FileInputStream('巴新EDEVU水库数据处理表.xlsx');
            XSSFWorkbook workbook = new XSSFWorkbook(fileIn);
            XSSFSheet sheet1 = workbook.getSheet('P9');
            XSSFSheet sheet2 = workbook.getSheet('P10');

            // 创建折线图
            XSSFDrawing drawing = sheet2.createDrawingPatriarch();
            XSSFClientAnchor anchor = drawing.createAnchor(0, 0, 0, 0, 3, 9, 9, 19);
            XSSFChart chart = drawing.createChart(anchor);
            chart.setTitleText('数据折线图');
            chart.setTitleOverlay(false);

            // 设置图例位置
            XDDFChartLegend legend = chart.getOrAddLegend();
            legend.setPosition(LegendPosition.BOTTOM);

            // 创建数据系列
            XDDFCategoryAxis bottomAxis = chart.createCategoryAxis(AxisPosition.BOTTOM);
            XDDFValueAxis leftAxis = chart.createValueAxis(AxisPosition.LEFT);
            leftAxis.setCrosses(AxisCrosses.AUTO_ZERO);

            // 创建数据源
            XDDFNumericalDataSource<Double> xs1 = XDDFDataSourcesFactory.fromNumericCellRange(sheet1, new CellRangeAddress(11, 85, 0, 0));
            XDDFNumericalDataSource<Double> ys1 = XDDFDataSourcesFactory.fromNumericCellRange(sheet1, new CellRangeAddress(11, 85, 2, 2));
            XDDFNumericalDataSource<Double> xs2 = XDDFDataSourcesFactory.fromNumericCellRange(sheet2, new CellRangeAddress(11, 139, 0, 0));
            XDDFNumericalDataSource<Double> ys2 = XDDFDataSourcesFactory.fromNumericCellRange(sheet2, new CellRangeAddress(11, 139, 2, 2));

            // 添加数据系列
            XDDFLineChartData data = (XDDFLineChartData) chart.createData(ChartTypes.LINE, bottomAxis, leftAxis);
            XDDFLineChartData.Series series1 = (XDDFLineChartData.Series) data.addSeries(xs1, ys1);
            series1.setTitle('P9数据折线图', null);
            XDDFLineChartData.Series series2 = (XDDFLineChartData.Series) data.addSeries(xs2, ys2);
            series2.setTitle('P10数据折线图', null);

            // 绘制图表
            chart.plot(data);

            // 创建'温度'工作表
            XSSFSheet sheet3 = workbook.createSheet('温度');

            // 将图表复制到'温度'工作表
            XSSFDrawing drawing2 = sheet3.createDrawingPatriarch();
            XSSFClientAnchor anchor2 = drawing2.createAnchor(0, 0, 0, 0, 0, 0, 10, 20);
            XSSFChart chart2 = drawing2.createChart(anchor2);
            chart2.setTitleText('数据折线图');
            chart2.setTitleOverlay(false);
            XDDFCategoryAxis bottomAxis2 = chart2.createCategoryAxis(AxisPosition.BOTTOM);
            XDDFValueAxis leftAxis2 = chart2.createValueAxis(AxisPosition.LEFT);
            leftAxis2.setCrosses(AxisCrosses.AUTO_ZERO);
            XDDFLineChartData data2 = (XDDFLineChartData) chart2.createData(ChartTypes.LINE, bottomAxis2, leftAxis2);
            data2.addSeries(xs1, ys1);
            data2.addSeries(xs2, ys2);
            chart2.plot(data2);

            // 将XSSFChart对象转换为XDDFChart对象
            XDDFChart xddfChart2 = (XDDFChart) chart2;

            // 绘制图表到图片
            BufferedImage image = new BufferedImage(800, 600, BufferedImage.TYPE_INT_RGB);
            Graphics2D graphics = image.createGraphics();
            xddfChart2.draw(graphics, new Rectangle(0, 0, 800, 600));

            // 保存图片文件
            ImageIO.write(image, 'png', new File('折线图.png'));

            // 保存Excel文件
            FileOutputStream fileOut = new FileOutputStream('巴新EDEVU水库数据处理表.xlsx');
            workbook.write(fileOut);
            fileOut.close();

            System.out.println('折线图已创建并保存到Excel文件和图片中。');

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

3. 代码解析

1. 导入必要的类： 导入Apache POI库相关的类，包括XSSFWorkbook、XSSFSheet、XSSFChart、XDDFChart、XDDFLineChartData等。
2. 读取Excel文件： 使用FileInputStream读取Excel文件，并创建XSSFWorkbook对象。
3. 创建折线图： 使用createDrawingPatriarch()方法创建XSSFDrawing对象，然后使用createChart()方法创建XSSFChart对象。
4. 设置图表标题和图例： 使用setTitleText()方法设置图表标题，使用getOrAddLegend()方法获取图例对象，并使用setPosition()方法设置图例位置。
5. 创建数据系列： 使用createCategoryAxis()方法创建水平轴（X轴），使用createValueAxis()方法创建垂直轴（Y轴）。
6. 创建数据源： 使用fromNumericCellRange()方法从Excel表格中获取数据，创建XDDFNumericalDataSource对象。
7. 添加数据系列： 使用addSeries()方法将数据源添加到XDDFLineChartData对象中。
8. 绘制图表： 使用plot()方法将数据系列绘制到图表上。
9. 将图表复制到其他工作表： 使用createSheet()方法创建新的工作表，并使用createDrawingPatriarch()、createAnchor()和createChart()方法将图表复制到新的工作表中。
10. 绘制图表到图片： 将XSSFChart对象转换为XDDFChart对象，然后使用draw()方法将图表绘制到BufferedImage对象上。
11. 保存图片文件： 使用ImageIO.write()方法将BufferedImage对象保存为图片文件。
12. 保存Excel文件： 使用FileOutputStream将XSSFWorkbook对象保存到Excel文件。

4. 运行代码

运行代码后，您将看到一个名为'折线图.png'的图片文件，该图片包含您从Excel数据创建的折线图。同时，您还会看到一个名为'巴新EDEVU水库数据处理表.xlsx'的Excel文件，其中包含一个名为'温度'的工作表，该工作表也包含相同的折线图。

5. 总结

本教程展示了如何使用Apache POI库在Java程序中生成Excel折线图，并保存为图片文件。您可以根据自己的需要修改代码，以创建更加复杂的图表。