使用EasyExcel对excel数据进行相似度判断

1. [/code][code]@Data
2. public class ExeclDto {
3.     /**
4.      * execl表
5.      */
6.     private String filename;
7.     /**
8.      * 需要匹配的工作表名
9.      */
10.     private String name1;
11.     /**
12.      * 需要匹配的工作表名
13.      */
14.     private String name2;
15. }

复制代码

1. [/code] 
2. [code]@SpringBootTest
3. @Slf4j
4. class CmmcysApplicationTests {
6.     @Test
7.     public void testReadExcel() throws IOException, InvalidFormatException {
8.         // 读取的excel文件路径
9.         String filename = "src/main/resources/static/xlsx/车型名称映射.xlsx";
10.         String filename_new = "src/main/resources/static/xlsx/结果表.xlsx";
11.         // 判断filename是否为空
12.         if (!StringUtils.isEmpty(filename)) {
13.             //读取excel
14.             ExeclDto execlDto = new ExeclDto();
15.             execlDto.setFilename(filename);
16.             execlDto.setName1("A");
17.             execlDto.setName2("B");
18.             Map<String, String> map = readExcel(execlDto);
19.             //如果map非空就写入
20.             if (!map.isEmpty()) {
21.                 List list = new ArrayList<>();
22.                 for (Map.Entry<String, String> entry : map.entrySet()) {
23.                     CarDto carDto = new CarDto();
24.                     carDto.setName_a(entry.getKey());
25.                     carDto.setName_b(entry.getValue());
26.                     list.add(carDto);
27.                     log.info(entry.getKey() + " ；" + entry.getValue());
28.                 }
29.                 EasyExcel.write(filename_new, CarDto.class).sheet("结果表").doWrite(list);
30.             }else {
31.                 System.out.println("文件没有该工作表，请重新检查上传");
32.             }
33.         } else {
34.             System.out.println("文件为空，请重新上传");
35.         }
36.     }
39.     public Map<String,String> readExcel(ExeclDto execlDto){
40.         // 读取的excel文件路径
41.         String filename = execlDto.getFilename();
42.         // 读取excel
43.         File file = new File(filename);
44.         Workbook sheets = null;
45.         try {
46.             sheets = WorkbookFactory.create(file);
47.         } catch (IOException e) {
48.             e.printStackTrace();
49.         } catch (InvalidFormatException e) {
50.             e.printStackTrace();
51.         }
52.         Sheet sheetAt1 = sheets.getSheet(execlDto.getName1());
53.         Sheet sheetAt2 = sheets.getSheet(execlDto.getName2());
54.         //判断excel表中是否有该工作表
55.         if (sheetAt1 !=null && sheetAt2 !=null) {
56.             //采用LinkedHashMap保证数据的顺序性
57.             Map<String, Double> map1 = new LinkedHashMap<>();
58.             Map<String, String> map2 = new LinkedHashMap<>();
59.             //循环工作表行
60.             for (Row row1 : sheetAt1) {
61.                 //获取工作表列值
62.                 String stringCellValue = row1.getCell(0).getStringCellValue();
63.                 //设置string类型初始阈值
64.                 map1.put("bz", 0.0);
65.                 //循环需要比较的工作表列值
66.                 for (Row row2 : sheetAt2) {
67.                     //获取工作表列值
68.                     String stringCellValue1 = row2.getCell(0).getStringCellValue();
69.                     //判断车辆款式，获取字符串位置
70.                     if (stringCellValue.contains("款") && stringCellValue1.contains("款")) {
71.                         int i = stringCellValue.indexOf("款");
72.                         int b = stringCellValue1.indexOf("款");
73.                         //进行款式截取并比较款式是否一致
74.                         if (stringCellValue.substring(i - 4, i).equals(stringCellValue1.substring(b - 4, b))) {
75.                             //采用String里面的方法进行相似度取值
76.                             double similarity = StringUtils.getJaroWinklerDistance(stringCellValue, stringCellValue1);
77.                             //进行相似度比较
78.                             if (similarity > map1.get("bz")) {
79.                                 //相似度高的替换低的
80.                                 map1.put("bz", similarity);
81.                                 //存放进map集合
82.                                 map2.put(stringCellValue, stringCellValue1);
83.                             }
84.                         }
85.                     }
86.                 }
87.             }
88.             return map2;
89.         }else {
90.             return null;
91.         }
92.     }
93. }

复制代码

 

 

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！