// 将查询结果保存到csv文件 - OutlEts - 沸点 - 掘金

出发口

#码上掘金#
// 将查询结果保存到csv文件
fn save_search_result(filename: &str, search_result: Vec<Vec<String>>) {
let f = std::fs::File::create(filename).expect("Error creating file");
let mut wtr = csv::Writer::from_writer(BufWriter::new(f));

for row in search_result {
wtr.write_record(row).expect("Error writing record");
}
}

fn search_data(filename: &str, keyword: &str) -> Vec<Vec<String>> {
let csv_data = read_csv(filename);
let mut search_result = Vec::new();

for row in csv_data {
for value in row {
if value.contains(keyword) {
search_result.push(row);
break;
}
}
}

search_result
}

fn main() {
let filename = "data.csv";
let keyword = "apple";

let search_result = search_data(filename, keyword);
save_search_result("result.csv", search_result);
}

展开

赞过