抟扶

  博客园 :: 首页 :: 博问 :: 闪存 :: 新随笔 :: 联系 :: 订阅 订阅 :: 管理 ::
package dwzx.com.get;


import java.io.BufferedReader; 
import java.io.File;
import java.io.FileReader;
import java.io.IOException;
import java.util.ArrayList;
import java.util.StringTokenizer;

//解析指定目录下的所有csv文件,返回三维数组数据
public class GetCsvData {
    
    public static final int COLUMN_NUM = 9;//csv文件列数
    
    //获取指定csv文件数据,存入二维数组并返回
    public static String[][] getCsvDataNew(String filePath) throws IOException{
         BufferedReader br = new BufferedReader(new FileReader(new File(filePath)));
         String line = "";
         ArrayList<String[]> lineList = new ArrayList<String[]>(); 
        // Read a single line from the file until there are no more lines to read
         while((line = br.readLine()) != null) { 
             StringTokenizer st = new StringTokenizer(line, ","); // 以逗号作为分隔符
             String[] currCol = new String[COLUMN_NUM]; // Each currCol has 9 fields, so we need room for the 8 tokens.
             for(int i = 0; i < COLUMN_NUM; i++) { // For each token in the line that we've read:
                 //先判断是否还有待读取数据,防止溢出
                 if(st.hasMoreTokens()){
                     currCol[i]  = st.nextToken(); 
                 }
             
             }
             lineList.add(currCol); 
         }
         
        String[][] str = new String[lineList.size()][9];
        for(int i = 0; i < lineList.size(); i++) {
              for(int j = 0; j < 9; j++) {
               str[i][j] = lineList.get(i)[j];
               //System.out.println(str[i][x]);
              }
        }
        
        br.close();
        
        return str;
    }

    //获取目录当前路径下所有csv文件的绝对路径
    public static ArrayList<String> getFileList(String dirPath){
        File dir = new File(dirPath);
        File[] fileList = dir.listFiles();
        ArrayList<String> strList = new ArrayList<String>();
        for(File f:fileList){
                            
            if((f.isFile()) 
                    && (".csv".equals(
                            f.getName().
                            substring(
                                    f.getName().lastIndexOf("."), 
                                    f.getName().length())))){
                strList.add(f.getAbsolutePath());
                
            }
        }
        
        
        return strList;

    }
    
    //获取指定目录下所有csv文件数据,存入三维数组并返回
    public static String[][][] getAllCsvData(String dir) throws IOException{
        //ArrayList<String> list = new ArrayList<String>();
        ArrayList<String> csvList = getFileList(dir);
        
        String[][][] s3 = new String[csvList.size()][][];
        for(int i = 0; i < csvList.size(); i ++){
            
            s3[i] = (getCsvDataNew(csvList.get(i)));
                
        }
        return s3;
    }
    
    
    
    public static void main(String[] args) throws IOException {
        
        String[][][] s = getAllCsvData("testdata\\");
        for(int i = 0; i < s.length; i ++)
            for(int j = 0; j < s[i].length; j ++)
                for(int k = 0; k < s[i][j].length; k ++){
                    System.out.println("s["+i+"]"+"["+j+"]"+"["+k+"]: "+s[i][j][k]);
                }
    
            }
    
}

 

posted on 2015-11-18 18:40  抟扶  阅读(3498)  评论(1编辑  收藏  举报