记录下使用Apache POI怎么读取Excel数据

今天学习了下使用TestNG、Apache POI和Excel文件进行数据驱动测试,当在写使用POI读取Excel数据时遇到了几个坑,所以在这里记录下!

首先记录下一个获取Excel数据的方法:

有些地方加了一些注释;

package test;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
import java.util.Locale;

import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.DataFormatter;
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.xssf.usermodel.XSSFWorkbook;

public class Test001 {
	public static void main(String[] args) throws IOException   {
		Test001  t =new Test001();
		Object[][ ]  ee= t.getTestData("D:\\", "qq.xlsx", "文件1");
		
		for(Object a[]:ee) {
			
			for(Object c:a) {
				System.out.print(c+"  ");
			}
			System.out.println();
			
		}
	}
	
	
	
	public  Object [][] getTestData (String filePath,String fileName,String sheetName) {


		File file = new File(filePath + "/"+ fileName);

		FileInputStream inputStream=null;
		try {
			inputStream = new FileInputStream(file);
		} catch (FileNotFoundException e) {
			// TODO 自动生成的 catch 块
			e.printStackTrace();
		}
		 DataFormatter formatter = new DataFormatter(Locale.US);  

		Workbook Workbook = null;

		String fileExtensionName = fileName.substring(fileName.indexOf("."));

		try {
			if(fileExtensionName.equals(".xlsx")){
			Workbook = new XSSFWorkbook(inputStream);
			}else if (fileExtensionName.equals(".xls")){
			Workbook = new HSSFWorkbook(inputStream);	
			}
		} catch (IOException e) {
			// TODO 自动生成的 catch 块
			e.printStackTrace();
		}


		Sheet Sheet = Workbook.getSheet(sheetName);

		int rowCount = (Sheet.getLastRowNum() - Sheet.getFirstRowNum())+1;


		List<Object[]> records = new ArrayList <Object[]>();

		for (int i=1;i<rowCount;i++){  //因为excel的行列是从0开始的,0行是数据列名称
		Row row = Sheet.getRow(i);   //用getRow()获取行对象

		String fields[] = new String[row.getLastCellNum()];  //用getLastCellNum()动态声明一个数组
		for(int j = 0;j<row.getLastCellNum();j++){
			
			//最初写法*************报错   Cannot get a text value from a numeric cell
			//fields[j]=row.getCell(j).getStringCellValue();
			
//			row.getCell(j).setCellType(row.getCell(j).CELL_TYPE_STRING);
//			fields[j]=row.getCell(j).getStringCellValue();   //不建议使用这种方法!
			
			
			fields[j] =formatter.formatCellValue(row.getCell(j));  //把elecel每列的元素保存到动态数组
	
		}
		records.add(fields);  //把数组加入到集合中
		}

		
		//把集合转换成一个二维数组
		//这里有一点点不好想的地方,仔细想想fields 数组是在集合中是怎么存储的??????

		Object[][] results = new Object[records.size()][];
		
		
		
		for (int i=0;i<records.size();i++){
		results[i]=records.get(i);
		}
		return results;

		}

		}

第一坑:

sheetName 是打开的文件显示的什么名字就写什么,不是Sheet1.

如果写错了会显示: