如何获取Rust-Polars DataFrame的列数？

在Rust的Polars库中，你可以通过以下三种简洁的方式获取DataFrame的列数：

方法1：使用width()方法 #

DataFrame自带的width()方法会直接返回列的数量，这是最直接的实现方式：

use polars::prelude::*;

fn main() {
    let df = df!("1" => &[1, 2, 3, 4, 5,6], "2" => &[2, 3, 4, 5, 6, 6]).unwrap();
    let column_count = df.width();
    println!("列数：{}", column_count); // 输出结果：列数：2
}

方法2：通过列名集合的长度获取 #

调用columns()方法获取所有列名的数组，再取数组的长度即可得到列数：

use polars::prelude::*;

fn main() {
    let df = df!("1" => &[1, 2, 3, 4, 5,6], "2" => &[2, 3, 4, 5, 6, 6]).unwrap();
    let column_count = df.columns().len();
    println!("列数：{}", column_count); // 输出结果：列数：2
}

方法3：从shape元组中提取 #

调用shape()方法会返回一个包含(行数, 列数)的元组，通过索引1即可取出列数：

use polars::prelude::*;

fn main() {
    let df = df!("1" => &[1, 2, 3, 4, 5,6], "2" => &[2, 3, 4, 5, 6, 6]).unwrap();
    let (_, column_count) = df.shape();
    println!("列数：{}", column_count); // 输出结果：列数：2
}

以上三种方法都能准确得到目标列数，其中width()方法是最简洁直观的选择。

内容的提问来源于stack exchange，提问作者user459872