java和python读写CSV文件

java的读写操作

1.读取CSV文件

try {    
      BufferedReader buffeReader = new BufferedReader(new FileReader("test.csv")); 
      buffeReader.readLine();//读取标题行
      String data = null;    
      while((data = buffeReader.readLine())!=null){ //循环读取每一行数据   
           String info[] = data.split(",");//CSV格式文件为逗号分隔符文件 ，根据“，”区分出每一列数据
       }    
} catch (Exception e) {    
       e.printStackTrace();    
}

2.写入CSV文件

public void saveInfo() {
   File csvFile = null;
   BufferedWriter csvFileOutputStream = null;
   List<String> csvDataList = new ArrayList<String>();
   csvDataList.add("test0");
   csvDataList.add("test1");

   try{
       csvFile = new File("test.csv");
       if (!csvFile.exists()) {
           csvFile.createNewFile();
       }
       csvFileOutputStream = new BufferedWriter(new OutputStreamWriter(new FileOutputStream(csvFile, true), StandardCharsets.UTF_8), 1024);
       writeRow(csvDataList, csvFileOutputStream);
       csvFileOutputStream.newLine();
   } catch (IOException e) {
       e.printStackTrace();
   } finally {
       try {
           if (csvFileOutputStream != null) {
               csvFileOutputStream.close();
           }
       } catch (IOException e) {
           e.printStackTrace();
       }
   }
}

private void writeRow(List<String> row, BufferedWriter csvWriter) throws IOException {
   int i = 0;
   for (String data : row) {
       csvWriter.write(data);
       if (i != row.size() - 1) {
           csvWriter.write(",");
       }
       i++;
   }
   csvWriter.newLine();
}

pyhton读写CSV文件

1.读取csv文件

import pandas as pd

file = pd.read_csv('test.csv')
df = pd.DataFrame(file)

for i in range(len(df))://按行读取csv文件
    document = df[i:i+1]
    data0 = document['data0'][i] //根据列名，案列得出每列数据
    data1 = document['data1'][i]
    print(data0,data1,'\n')//打印每行数据

2.写入csv文件

import pandas as pd

dict = {'data0': 95, 'data1': 82}

df = pd.DataFrame.from_dict(dict)
df.to_csv('test.csv')