Início / coding / Perguntas / 79578841
Accepted
zwh82
Asked: 2025-04-17 16:48:31 +0800 CST

Como usar Polars para ler colunas específicas de um arquivo CSV

  • 772

Tenho um arquivo muito grande gerado por outras ferramentas, mas não preciso de todas as informações; apenas algumas colunas são suficientes. Quando uso o Python Pandas para ler, consigo especificar as colunas necessárias, mas não sei como o Rust implementa isso.

Obrigado.

Espero que o Rust possa atingir a mesma funcionalidade que o Python Pandas.

data = pd.read_csv(file, sep='\t', header=None, usecols=[0,1,5])
rust

1 respostas

  1. Best Answer

    Presumo que você queira usar rust-polars. É assim que você poderia conseguir o mesmo usando ferrugem.

    Também adicionei o comentário para entender o que acontece com cada etapa.

    use polars::prelude::*;

fn main() {
    let path = "<PATH_TO_THE_FILE>";
    // If there are no headers, polars automatically choose "column_1, column_2 etc"
    let columns_to_select = ["column_1".into(), "column_2".into()];

    let df = CsvReadOptions::default()
        .with_has_header(false) // equivalent to `header=None` in pandas 
        .map_parse_options(|parse_options| parse_options.with_separator(b'\t')) // use custom separator. equivalent to `sep=\t` in pandas 
        .with_columns(Some(Arc::new(columns_to_select))) // select the columns. equivalent to `usecols=[1, 2]` in pandas
        .try_into_reader_with_file_path(Some(path.into())) // specify the file path
        .unwrap()
        .finish()
        .unwrap();
    println!("{:?}", df);
}
    • 0

relate perguntas