Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- import java.io.BufferedReader;
- import java.io.BufferedWriter;
- import java.io.File;
- import java.io.FileNotFoundException;
- import java.io.FileOutputStream;
- import java.io.FileReader;
- import java.io.FileWriter;
- import java.io.IOException;
- import java.io.PrintWriter;
- import java.util.ArrayList;
- import java.util.Arrays;
- import java.util.Iterator;
- import java.util.Scanner;
- public class big_data {
- public static void main(String[] params) throws Exception {
- int id=Integer.parseInt(params[3]);
- String csvFile = params[0]+"cleanData.csv";
- int sample_size=60;
- int samples_collected=0;
- String line = "";
- String cvsSplitBy = ";";
- boolean isNotHeader=true;
- int id_riadku=0;
- if(id==0)
- isNotHeader=false;
- else
- isNotHeader=true;
- PrintWriter bigData = new PrintWriter(new FileOutputStream(new File(params[1]+"bigData.csv"), true /* append = true */));
- try (BufferedReader br = new BufferedReader(new FileReader(csvFile))) {
- while ((line = br.readLine()) != null) {
- samples_collected=0;
- StringBuilder sb = new StringBuilder();
- String[] input = line.split(cvsSplitBy);
- if(id_riadku!=0||isNotHeader==false)
- for(int i=1;i<input.length;i++){
- sb.append(input[i]+";");
- }
- if(isNotHeader)
- sb.append(params[3]+";");
- else
- sb.append("user_id;");
- if(params[2].equals("Kid")&&(isNotHeader)) sb.append("1;");
- if(params[2].equals("Adult")&&(isNotHeader)) sb.append("0;");
- if(isNotHeader==false) sb.append("is_kid;");
- sb.append("\n");
- if(isNotHeader==false&&id==0) bigData.write(sb.toString());
- if(isNotHeader&&id_riadku!=0&samples_collected<=sample_size){
- bigData.write(sb.toString());
- samples_collected++;
- }
- isNotHeader=true;
- id_riadku++;
- }
- }
- samples_collected=0;
- bigData.close();
- System.out.println("done!");
- }
- }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement