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:
 */
@Slf4j
public 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("不等");
            }
        }

    }

}