如何使用Apache Arrow Rust Crate显示UTF8列内容？

如何解决如何使用Apache Arrow Rust Crate显示UTF8列内容？

我根据rust的apache :: arrow实现的示例和文档成功加载了实木复合地板文件。

    use parquet::arrow::{ParquetFileArrowReader,ArrowReader};
    use std::rc::Rc;
    use arrow::record_batch::RecordBatchReader;

    let file = File::open(&Path::new("./path_to/file.parquet")).unwrap();
    let file_reader = SerializedFileReader::new(file).unwrap();
    let mut arrow_reader = ParquetFileArrowReader::new(Rc::new(file_reader));

    println!("Converted arrow schema is: {}",arrow_reader.get_schema().unwrap());

    let mut record_batch_reader = arrow_reader.get_record_reader(2048).unwrap();

我能够显示每个批次的名称和列的类型：

    loop {
       let record_batch = record_batch_reader.next_batch().unwrap().unwrap();
       if record_batch.num_rows() > 0 {
           println!("Schema: {}.",record_batch.schema());
       }
    }

但是我对如何显示列的内容很困惑。如何检索第一列的内容并打印出来？

解决方法

最新版本的Apache箭头接缝具有prettifyer类。不幸的是，这不在最后一个可用的软件包（1.0.1）中。

use arrow::util::pretty;
pretty::print_batches(&batch);

手动操作是通过向下转换。

// For an int:
let col = batch.column(0).as_any().downcast_ref::<arrow::array::Int32Array>();

// For a Utf8 string:
let col = batch.column(0).as_any().downcast_ref::<arrow::array::StringArray>();

然后您可以简单地打印它：

println!("Columns: {:?}.",col);