使用EasyExcel對excel資料進行相似度判斷

@Data
public class ExeclDto {
    /**
     * execl表
     */
    private String filename;
    /**
     * 需要匹配的作業表名
     */
    private String name1;
    /**
     * 需要匹配的作業表名
     */
    private String name2;
}

@SpringBootTest
@Slf4j
class CmmcysApplicationTests {

    @Test
    public void testReadExcel() throws IOException, InvalidFormatException {
        // 讀取的excel檔案路徑
        String filename = "src/main/resources/static/xlsx/車型名稱映射.xlsx";
        String filename_new = "src/main/resources/static/xlsx/結果表.xlsx";
        // 判斷filename是否為空
        if (!StringUtils.isEmpty(filename)) {
            //讀取excel
            ExeclDto execlDto = new ExeclDto();
            execlDto.setFilename(filename);
            execlDto.setName1("A");
            execlDto.setName2("B");
            Map<String, String> map = readExcel(execlDto);
            //如果map非空就寫入
            if (!map.isEmpty()) {
                List list = new ArrayList<>();
                for (Map.Entry<String, String> entry : map.entrySet()) {
                    CarDto carDto = new CarDto();
                    carDto.setName_a(entry.getKey());
                    carDto.setName_b(entry.getValue());
                    list.add(carDto);
                    log.info(entry.getKey() + " ；" + entry.getValue());
                }
                EasyExcel.write(filename_new, CarDto.class).sheet("結果表").doWrite(list);
            }else {
                System.out.println("檔案沒有該作業表，請重新檢查上傳");
            }
        } else {
            System.out.println("檔案為空，請重新上傳");
        }
    }


    public Map<String,String> readExcel(ExeclDto execlDto){
        // 讀取的excel檔案路徑
        String filename = execlDto.getFilename();
        // 讀取excel
        File file = new File(filename);
        Workbook sheets = null;
        try {
            sheets = WorkbookFactory.create(file);
        } catch (IOException e) {
            e.printStackTrace();
        } catch (InvalidFormatException e) {
            e.printStackTrace();
        }
        Sheet sheetAt1 = sheets.getSheet(execlDto.getName1());
        Sheet sheetAt2 = sheets.getSheet(execlDto.getName2());
        //判斷excel表中是否有該作業表
        if (sheetAt1 !=null && sheetAt2 !=null) {
            //采用LinkedHashMap保證資料的順序性
            Map<String, Double> map1 = new LinkedHashMap<>();
            Map<String, String> map2 = new LinkedHashMap<>();
            //回圈作業表行
            for (Row row1 : sheetAt1) {
                //獲取作業表列值
                String stringCellValue = https://www.cnblogs.com/kxxz/archive/2023/06/20/row1.getCell(0).getStringCellValue();
                //設定string型別初始閾值
                map1.put("bz", 0.0);
                //回圈需要比較的作業表列值
                for (Row row2 : sheetAt2) {
                    //獲取作業表列值
                    String stringCellValue1 = row2.getCell(0).getStringCellValue();
                    //判斷車輛款式，獲取字串位置
                    if (stringCellValue.contains("款") && stringCellValue1.contains("款")) {
                        int i = stringCellValue.indexOf("款");
                        int b = stringCellValue1.indexOf("款");
                        //進行款式截取并比較款式是否一致
                        if (stringCellValue.substring(i - 4, i).equals(stringCellValue1.substring(b - 4, b))) {
                            //采用String里面的方法進行相似度取值
                            double similarity = StringUtils.getJaroWinklerDistance(stringCellValue, stringCellValue1);
                            //進行相似度比較
                            if (similarity > map1.get("bz")) {
                                //相似度高的替換低的
                                map1.put("bz", similarity);
                                //存放進map集合
                                map2.put(stringCellValue, stringCellValue1);
                            }
                        }
                    }
                }
            }
            return map2;
        }else {
            return null;
        }
    }
}

標籤：其他

上一篇：【whale-starry-stl】01天 list學習筆記

下一篇：返回列表