apache poi拆分excel表格
前几天遇到一个需求，需要把一个有几万行有规则数据的excel文件拆分成N个excel文件,例如以下excel表格中包括以张三，李四，王五的各科考试成绩，每个学生的学号是唯一的，现在要求把每个人的成绩数据分开来生成独立的表格数据，并以学号作为文件名。思路: 使用HashMap，用每个工作表作为值，用学号作为键。遍历原始工作表中的每一行，以学号为键去HashMap对象中取工作表，如果不存在，那么主新建一个工作表，写入此行数据并保存工作表在HashMap对象中。如果以当前行的学号作为键可以取到工作表，则取出工作表，并将当前行写入工作表，并重新保存工作表到HashMap对象中。最后遍历HashMap对象，重新生成Excel文件。
以下为实现代码
import java.io.File;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.FileInputStream;
import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;
import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.DateUtil;
import org.apache.poi.ss.usermodel.Row;
import org.apache.poi.ss.usermodel.Sheet;
import org.apache.poi.ss.usermodel.Workbook;
import org.apache.poi.ss.usermodel.WorkbookFactory;
import org.apache.poi.xssf.usermodel.XSSFCellStyle;
import org.apache.poi.xssf.usermodel.XSSFDataFormat;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
public class ExcelSplit {
public static void main(String[] args){
try {
System.out.println("开始拆分.....");
Map<String, XSSFWorkbook> map=getSplitMap("D:/xxxx.xlsx");//得到拆分后的子文件存储对象
createSplitXSSFWorkbook(map, "D:/splitdone/201404", "2014.04");//遍历对象生成的拆分文件
System.out.println("拆分结束,文件被拆分为"+map.size()+"个文件.");
} catch (Exception e) {
e.printStackTrace();
}
}
//将第一列的值作为键值,将一个文件拆分为多个文件
public static Map<String,XSSFWorkbook> getSplitMap(String fileName) throws Exception{
Map<String,XSSFWorkbook> map=new HashMap<String,XSSFWorkbook>();
InputStream is = new FileInputStream(new File(fileName));
//根据输入流创建Workbook对象
Workbook wb = WorkbookFactory.create(is);
//get到Sheet对象
Sheet sheet = wb.getSheetAt(0);
Row titleRow=null;
//这个必须用接口
int i=0;
for(Row row : sheet){//遍历每一行
if(i==0){
titleRow=row;//得到标题行
}else{
Cell keyCell=row.getCell(0);
String key=keyCell.getRichStringCellValue().toString();
XSSFWorkbook tempWorkbook=map.get(key);
if(tempWorkbook==null){//如果以当前行第一列值作为键值取不到工作表
tempWorkbook= new XSSFWorkbook();
Sheet tempSheet=tempWorkbook.createSheet();
Row firstRow=tempSheet.createRow(0);
for(short k=0;k<titleRow.getLastCellNum();k++){//为每个子文件创建标题
Cell c=titleRow.getCell(k);
Cell newcell=firstRow.createCell(k);
newcell.setCellValue(c.getStringCellValue());
}
map.put(key,tempWorkbook);
}
Sheet secSheet=tempWorkbook.getSheetAt(0);
Row secRow=secSheet.createRow(secSheet.getLastRowNum()+1);
for(short m=0;m<row.getLastCellNum();m++){
Cell newcell=secRow.createCell(m);
setCellValue(newcell,row.getCell(m),tempWorkbook);
}
map.put(key,tempWorkbook);
}
i=i+1;//行数加一
}
return map;
}
//创建文件
public static void createSplitXSSFWorkbook(Map<String, XSSFWorkbook> map,String savePath,String month)
throws IOException {
Iterator iter = map.entrySet().iterator();
while (iter.hasNext()) {
Map.Entry entry = (Map.Entry) iter.next();
String key = (String) entry.getKey();
XSSFWorkbook val = (XSSFWorkbook) entry.getValue();
File filePath=new File(savePath);
if(!filePath.exists()){
System.out.println("存放目录不存在,自动为您创建存放目录.");
filePath.mkdir();
}
if(!filePath.isDirectory()){
System.out.println("无效文件目录");
return ;
}
File file=new File(savePath+"/"+key+"_"+month+".xlsx");
FileOutputStream fOut;// 新建输出文件流
try {
fOut = new FileOutputStream(file);
val.write(fOut); // 把相应的Excel工作薄存盘
fOut.flush();
fOut.close(); // 操作结束，关闭文件
} catch (FileNotFoundException e) {
System.out.println("找不到文件");
}
}
}
//将一个单元格的值赋给另一个单元格
public static void setCellValue(Cell newCell,Cell cell,XSSFWorkbook wb){
if(cell==null){
return;
}
switch(cell.getCellType()){
case Cell.CELL_TYPE_BOOLEAN:
newCell.setCellValue(cell.getBooleanCellValue());
break;
case Cell.CELL_TYPE_NUMERIC:
if(DateUtil.isCellDateFormatted(cell)){
XSSFCellStyle cellStyle =wb.createCellStyle();
XSSFDataFormat format= wb.createDataFormat();
cellStyle.setDataFormat(format.getFormat("yyyy/m/d"));
newCell.setCellStyle(cellStyle);
newCell.setCellValue(cell.getDateCellValue());
}else{
//读取数字
newCell.setCellValue(cell.getNumericCellValue());
}
break;
case Cell.CELL_TYPE_FORMULA:
newCell.setCellValue(cell.getCellFormula());
break;
case Cell.CELL_TYPE_STRING:
newCell.setCellValue(cell.getStringCellValue());
break;
}
}
}
有任何问题请联系qq 359709421
如果您觉得我的文章给了您帮助，请为我买一杯饮料吧！以下是我的支付宝，意思一下我将非常感激！
页:
[1]