poi讀取excel的列和刪除列

（各自根據具體的poi版本進行相應的替換便可）package com.br.loan.strategy.common.utils;import lombok.extern.slf4j.Slf4j;import org.apache.poi.hssf.usermodel.HSSFCell;import org.apache.poi.hssf.usermodel.HSSFWorkbook;import org.apache.poi.ss.usermodel.*;import org.apache.poi.xssf.streaming.SXSSFSheet;import org.apache.poi.xssf.streaming.SXSSFWorkbook;import org.apache.poi.xssf.usermodel.XSSFSheet;import org.apache.poi.xssf.usermodel.XSSFWorkbook;import java.io.*;import java.util.*;/** * @Company: * @Author: shaobin.fu * @Date: 2018/9/6 16:33 * @Description: */@Slf4jpublic class Excel {    //總行數    private int totalRows = 0;    //總列數    private int totalCells = 0;    //錯誤信息    private String errorInfo;    public Excel() {    }    //獲得總行數    public int getTotalRows() {        return totalRows;    }    //獲得總行數    public int getTotalCells() {        return totalCells;    }    //獲得錯誤信息    public String getErrorInfo() {        return errorInfo;    }    /**     * @描述：驗證excel文件     * @時間：2012-08-29 下午16:27:15     * @參數：@param filePath　文件完整路徑     * @參數：@return     * @返回值：boolean     */    public boolean validateExcel(String filePath) {        /** 檢查文件名是否爲空或者是不是Excel格式的文件 */        if (filePath == null || !(WDWUtil.isExcel2003(filePath) || WDWUtil.isExcel2007(filePath))) {            errorInfo = "文件名不是excel格式";            return false;        }        /** 檢查文件是否存在 */        File file = new File(filePath);        if (file == null || !file.exists()) {            errorInfo = "文件不存在";            return false;        }        return true;    }    /**     * @描述：根據文件名讀取excel文件     * @時間：2012-08-29 下午16:27:15     * @參數：@param filePath 文件完整路徑     * @參數：@return     * @返回值：List     */    public List<List<String>> read(String filePath) {        List<List<String>> dataLst = new ArrayList<List<String>>();        InputStream is = null;        try {            /** 驗證文件是否合法 */            if (!validateExcel(filePath)) {                System.out.println(errorInfo);                return null;            }            /** 判斷文件的類型，是2003仍是2007 */            boolean isExcel2003 = true;            if (WDWUtil.isExcel2007(filePath)) {                isExcel2003 = false;            }            /** 調用本類提供的根據流讀取的方法 */            File file = new File(filePath);            is = new FileInputStream(file);            dataLst = read(is, isExcel2003);            is.close();        } catch (Exception ex) {            ex.printStackTrace();        } finally {            if (is != null) {                try {                    is.close();                } catch (IOException e) {                    is = null;                    e.printStackTrace();                }            }        }        /** 返回最後讀取的結果 */        return dataLst;    }    /**     * @描述：根據文件名讀取excel文件，以列讀取     * @時間：2012-08-29 下午16:27:15     * @參數：@param filePath 文件完整路徑     * @參數：@return     * @返回值：List     */    /*public Map<Integer, Integer> readByColumn(String filePath) {        Map<Integer, Integer> dataLst = new HashMap<>();        InputStream is = null;        try {            *//** 驗證文件是否合法 *//*            if (!validateExcel(filePath)) {                System.out.println(errorInfo);                return null;            }            *//** 判斷文件的類型，是2003仍是2007 *//*            boolean isExcel2003 = true;            if (WDWUtil.isExcel2007(filePath)) {                isExcel2003 = false;            }            *//** 調用本類提供的根據流讀取的方法 *//*            File file = new File(filePath);            is = new FileInputStream(file);            dataLst = readByColumn(is, isExcel2003);            is.close();        } catch (Exception ex) {            ex.printStackTrace();        } finally {            if (is != null) {                try {                    is.close();                } catch (IOException e) {                    is = null;                    e.printStackTrace();                }            }        }        *//** 返回最後讀取的結果 *//*        return dataLst;    }*/    /**     * @描述：根據流讀取Excel文件     * @時間：2012-08-29 下午16:40:15     * @參數：@param inputStream     * @參數：@param isExcel2003     * @參數：@return     * @返回值：List     */    public List<List<String>> read(InputStream inputStream, boolean isExcel2003) {        List<List<String>> dataLst = null;        try {            /** 根據版本選擇建立Workbook的方式 */            Workbook wb = null;            if (isExcel2003) {                wb = new HSSFWorkbook(inputStream);            } else {                wb = new XSSFWorkbook(inputStream);            }            dataLst = read(wb);        } catch (IOException e) {            e.printStackTrace();        }        return dataLst;    }    /**     * @描述：根據流讀取Excel文件，以列讀取     * @時間：2012-08-29 下午16:40:15     * @參數：@param inputStream     * @參數：@param isExcel2003     * @參數：@return     * @返回值：List     */    public Map<Integer, Integer> readByColumn(InputStream inputStream, boolean isExcel2003) {        Map<Integer, Integer> dataLst = null;        try {            /** 根據版本選擇建立Workbook的方式 */            Workbook wb = null;            if (isExcel2003) {                wb = new HSSFWorkbook(inputStream);            } else {                wb = new XSSFWorkbook(inputStream);            }            dataLst = readByColumn(wb);        } catch (IOException e) {            e.printStackTrace();        }        return dataLst;    }    /**     * @描述：讀取數據，以行讀取     * @時間：2012-08-29 下午16:50:15     * @參數：@param Workbook     * @參數：@return     * @返回值：List<List<String>>     */    private List<List<String>> read(Workbook wb) {        List<List<String>> dataLst = new ArrayList<List<String>>();        /** 獲得第一個shell */        Sheet sheet = wb.getSheetAt(0);        /** 獲得Excel的行數 */        this.totalRows = sheet.getPhysicalNumberOfRows();        /** 獲得Excel的列數 */        if (this.totalRows >= 1 && sheet.getRow(0) != null) {            this.totalCells = sheet.getRow(0).getPhysicalNumberOfCells();        }        /** 循環Excel的行 */        for (int r = 0; r < this.totalRows; r++) {            Row row = sheet.getRow(r);            if (row == null) {                continue;            }            List<String> rowLst = new ArrayList<String>();            /** 循環Excel的列 */            for (int c = 0; c < this.getTotalCells(); c++) {                Cell cell = row.getCell(c);                String cellValue = "";                if (null != cell) {                    // 如下是判斷數據的類型                    switch (cell.getCellType()) {                        case HSSFCell.CELL_TYPE_NUMERIC: // 數字                            cellValue = cell.getNumericCellValue() + "";                            break;                        case HSSFCell.CELL_TYPE_STRING: // 字符串                            cellValue = cell.getStringCellValue();                            break;                        case HSSFCell.CELL_TYPE_BOOLEAN: // Boolean                            cellValue = cell.getBooleanCellValue() + "";                            break;                        case HSSFCell.CELL_TYPE_FORMULA: // 公式                            cellValue = cell.getCellFormula() + "";                            break;                        case HSSFCell.CELL_TYPE_BLANK: // 空值                            cellValue = "";                            break;                        case HSSFCell.CELL_TYPE_ERROR: // 故障                            cellValue = "非法字符";                            break;                        default:                            cellValue = "未知類型";                            break;                    }                }                rowLst.add(cellValue);            }            /** 保存第r行的第c列 */            dataLst.add(rowLst);        }        return dataLst;    }    /**     * @描述：讀取數據，以列讀取     * @時間：2012-08-29 下午16:50:15     * @參數：@param Workbook     * @參數：@return     * @返回值：List<List<String>>     */    public Map<Integer, Integer> readByColumn(SXSSFWorkbook wb) {        Map<Integer, Integer> dataMap = new LinkedHashMap<>();        /** 獲得第一個shell */        Sheet sheet = wb.getSheetAt(0);        /** 獲得Excel的行數 */        this.totalRows = sheet.getPhysicalNumberOfRows();        /** 獲得Excel的列數 */        if (this.totalRows >= 1 && sheet.getRow(0) != null) {            this.totalCells = sheet.getRow(0).getPhysicalNumberOfCells();        }        //從第九列開始        for (int lie = 9; lie <= totalCells; lie++) {            //設置爲1，表明不須要刪除，爲0，則刪除。            dataMap.put(lie, 0);        }        /** 循環Excel的行(不包括第一行) */        for (int r = 1; r < this.totalRows; r++) {            Row row = sheet.getRow(r);            if (row == null) {                continue;            }            /** 循環Excel的列 */            for (int c = 0; c < this.getTotalCells(); c++) {                Cell cell = row.getCell(c);                if (null != cell&&StringUtils.isNotEmpty(cell)) {                    dataMap.put(c+1, 1);                }            }        }        return dataMap;    }    /**     * @描述：讀取sheet數據，以列讀取     * @時間：2012-08-29 下午16:50:15     * @參數：@param Workbook     * @參數：@return     * @返回值：List<List<String>>     */    public Map<Integer, Integer> readSheetByColumn(Sheet sheet) {        Map<Integer, Integer> dataMap = new LinkedHashMap<>();        /** 獲得第一個shell */        /** 獲得Excel的行數 */        this.totalRows = sheet.getPhysicalNumberOfRows();        /** 獲得Excel的列數 */        if (this.totalRows >= 1 && sheet.getRow(0) != null) {            this.totalCells = sheet.getRow(0).getPhysicalNumberOfCells();        }        //從第九列開始        for (int lie = 9; lie <= totalCells; lie++) {            //設置爲1，表明不須要刪除，爲0，則刪除。            dataMap.put(lie, 0);        }        /** 循環Excel的行(不包括第一行) */        for (int r = 1; r < this.totalRows; r++) {            Row row = sheet.getRow(r);            if (row == null) {                continue;            }            /** 循環Excel的列 */            for (int c = 0; c < this.getTotalCells(); c++) {                Cell cell = row.getCell(c);                if (null != cell&&StringUtils.isNotEmpty(cell)) {                    dataMap.put(c+1, 1);                }            }        }        return dataMap;    }    /**     * 刪除列     * @param sheet     * @param columnToDelete     */    public void deleteColumn(SXSSFSheet sheet, int columnToDelete) {        for (int rId = 0; rId <= sheet.getLastRowNum(); rId++) {            Row row = sheet.getRow(rId);            for (int cID = columnToDelete; cID <= row.getLastCellNum(); cID++) {                Cell cOld = row.getCell(cID);                if (cOld != null) {                    row.removeCell(cOld);                }                Cell cNext = row.getCell(cID + 1);                if (cNext != null) {                    Cell cNew = row.createCell(cID, cNext.getCellTypeEnum());                    cloneCell(cNew, cNext);                    //Set the column width only on the first row.                    //Other wise the second row will overwrite the original column width set previously.                    if (rId == 0) {                        sheet.setColumnWidth(cID, sheet.getColumnWidth(cID + 1));                    }                }            }        }    }    /**     * 右邊列左移     * @param cNew     * @param cOld     */    private void cloneCell(Cell cNew, Cell cOld) {        cNew.setCellComment(cOld.getCellComment());        cNew.setCellStyle(cOld.getCellStyle());        if (CellType.BOOLEAN == cNew.getCellTypeEnum()) {            cNew.setCellValue(cOld.getBooleanCellValue());        } else if (CellType.NUMERIC == cNew.getCellTypeEnum()) {            cNew.setCellValue(cOld.getNumericCellValue());        } else if (CellType.STRING == cNew.getCellTypeEnum()) {            cNew.setCellValue(cOld.getStringCellValue());        } else if (CellType.ERROR == cNew.getCellTypeEnum()) {            cNew.setCellValue(cOld.getErrorCellValue());        } else if (CellType.FORMULA == cNew.getCellTypeEnum()) {            cNew.setCellValue(cOld.getCellFormula());        }    }    /**     * 讀取第一行的cell數據     */    public List<String> readFirstRow(SXSSFWorkbook wb) {        log.info("進入readFirstRow");        List<String> list = new ArrayList<>();        int totalColumn=0;        /** 獲得第一個shell */        Sheet sheet = wb.getSheetAt(0);        /** 獲得Excel的列數 */        if (sheet.getRow(0) != null) {            totalColumn = sheet.getRow(0).getPhysicalNumberOfCells();            log.info("totalColumn:{}",totalColumn);            for (int lie = 8; lie < totalColumn; lie++) {                log.info("lie:{}",sheet.getRow(0).getCell(lie).getStringCellValue());                list.add(sheet.getRow(0).getCell(lie).getStringCellValue());            }        }        log.info("readFirst的大小：{}",list.size());        return list;    }    /*public static void main(String[] args) {        File file = new File("C:\\Users\\Bairong\\Desktop\\sjfksj.xlsx");        Excel excel = new Excel();        try {            FileInputStream is = new FileInputStream(file);            Workbook wb = new XSSFWorkbook(is);            Map<Integer, Integer> stringIntegerMap = excel.readByColumn(wb);            Iterator<Map.Entry<Integer, Integer>> it = stringIntegerMap.entrySet().iterator();            while (it.hasNext()) {                Map.Entry<Integer, Integer> entry = it.next();                System.out.println("key= " + entry.getKey() + " and value= " + entry.getValue());                if (entry.getValue()==0){                    excel.deleteColumn(wb.getSheetAt(0),entry.getKey()-1);                }            }            FileOutputStream fileOut = new FileOutputStream(file);            wb.write(fileOut);            fileOut.close();            //excel.deleteColumn(sheet,1);            System.out.println("刪除成功");        } catch (Exception e) {            e.printStackTrace();        }    }*/}/** * @描述：工具類 * @時間：2012-08-29 下午16:30:40 */class WDWUtil {    /**     * @描述：是不是2003的excel，返回true是2003     * @時間：2012-08-29 下午16:29:11     * @參數：@param filePath　文件完整路徑     * @參數：@return     * @返回值：boolean     */    public static boolean isExcel2003(String filePath) {        return filePath.matches("^.+\\.(?i)(xls)$");    }    /**     * @描述：是不是2007的excel，返回true是2007     * @時間：2012-08-29 下午16:28:20     * @參數：@param filePath　文件完整路徑     * @參數：@return     * @返回值：boolean     */    public static boolean isExcel2007(String filePath) {        return filePath.matches("^.+\\.(?i)(xlsx)$");    }    public static void main(String[] args) {        List list1 = new ArrayList();        List list2 = new ArrayList();        list1.add("乾隆");        list1.add("李世民");        list2.add("康熙");        list2.add("李世");        for(int i=0;i<list1.size();i++){            if (list1.contains(list2.get(i))){                System.out.println("結果爲true");            }else {                System.out.println("不等");            }        }    }}