使用极性列表类型列-如何在条件下使用它们？

Question

我发现列表作为一种类型非常有用。不幸的是，它们都有详细的文档--如何生成它们，而不是如何访问内容本身:(

例如，我有一个名为CARS的专栏：

CARS
list
[f64]
======
[1,2,3]
[2,3]
[4]
[]

我想准备两个新的变量。

• CARF -列表中的第一个值
• CAR2 -1-如果value=2在第一位，2-如果2在任何地方，3-2根本不存在。

结果应该是：

CARF   CAR2
1      2   
2      1
4      3
Null   3

Answer 1

您可以使用when -> then -> otherwise表达式和arr命名空间。它由列表的实用程序组成。他们是记录在案。

下面是一个用例示例：

df = pl.DataFrame({
    "cars": [[1,2,3], [2,3],[4],[]]
})

df.select([
    pl.col("cars").arr.first().alias("cars_first"),
    pl.when(pl.col("cars").arr.first() == 2)
    .then(1)
    .when(pl.col("cars").arr.contains(2))
    .then(2)
    .otherwise(3).alias("cars_literal")
])

产出：

shape: (4, 2)
┌────────────┬──────────────┐
│ cars_first ┆ cars_literal │
│ ---        ┆ ---          │
│ i64        ┆ i32          │
╞════════════╪══════════════╡
│ 1          ┆ 2            │
├╌╌╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌╌╌╌╌┤
│ 2          ┆ 1            │
├╌╌╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌╌╌╌╌┤
│ 4          ┆ 3            │
├╌╌╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌╌╌╌╌┤
│ null       ┆ 3            │
└────────────┴──────────────┘