Início / coding / Perguntas / 79374797
Accepted
miroslaavi
Asked: 2025-01-21 22:29:57 +0800 CST

Como calcular a mediana horizontal

  • 772

Como calcular a mediana horizontal para colunas numéricas?

df = pl.DataFrame({"ABC":["foo", "bar", "foo"], "A":[1,2,3], "B":[2,1,None], "C":[1,2,3]})
print(df)

shape: (3, 4)
┌─────┬─────┬──────┬─────┐
│ ABC ┆ A   ┆ B    ┆ C   │
│ --- ┆ --- ┆ ---  ┆ --- │
│ str ┆ i64 ┆ i64  ┆ i64 │
╞═════╪═════╪══════╪═════╡
│ foo ┆ 1   ┆ 2    ┆ 1   │
│ bar ┆ 2   ┆ 1    ┆ 2   │
│ foo ┆ 3   ┆ null ┆ 3   │
└─────┴─────┴──────┴─────┘

Quero obter o mesmo que com o pl.mean_horizontal abaixo, mas obter mediana em vez da média. Não encontrei expressão existente para isso.

print(df.with_columns(pl.mean_horizontal(pl.col(pl.Int64)).alias("Horizontal Mean")))

shape: (3, 5)
┌─────┬─────┬──────┬─────┬─────────────────┐
│ ABC ┆ A   ┆ B    ┆ C   ┆ Horizontal Mean │
│ --- ┆ --- ┆ ---  ┆ --- ┆ ---             │
│ str ┆ i64 ┆ i64  ┆ i64 ┆ f64             │
╞═════╪═════╪══════╪═════╪═════════════════╡
│ foo ┆ 1   ┆ 2    ┆ 1   ┆ 1.333333        │
│ bar ┆ 2   ┆ 1    ┆ 2   ┆ 1.666667        │
│ foo ┆ 3   ┆ null ┆ 3   ┆ 3.0             │
└─────┴─────┴──────┴─────┴─────────────────┘
python

2 respostas

  1. Best Answer

    Não há median_horizontal()no momento, mas você pode usar

    df.with_columns(
    pl.concat_list(pl.col(pl.Int64)).list.median().alias("Horizontal Median")
)

    shape: (3, 5)
┌─────┬─────┬──────┬─────┬───────────────────┐
│ ABC ┆ A   ┆ B    ┆ C   ┆ Horizontal Median │
│ --- ┆ --- ┆ ---  ┆ --- ┆ ---               │
│ str ┆ i64 ┆ i64  ┆ i64 ┆ f64               │
╞═════╪═════╪══════╪═════╪═══════════════════╡
│ foo ┆ 1   ┆ 2    ┆ 1   ┆ 1.0               │
│ bar ┆ 2   ┆ 1    ┆ 2   ┆ 2.0               │
│ foo ┆ 3   ┆ null ┆ 3   ┆ 3.0               │
└─────┴─────┴──────┴─────┴───────────────────┘

    Ou você pode usar a integração numpy (mas isso provavelmente será mais lento):

    import numpy as np

df.with_columns(
    pl.Series("Horizontal Median", np.nanmedian(df.select(pl.col(pl.Int64)), axis=1))
)

    shape: (3, 5)
┌─────┬─────┬──────┬─────┬───────────────────┐
│ ABC ┆ A   ┆ B    ┆ C   ┆ Horizontal Median │
│ --- ┆ --- ┆ ---  ┆ --- ┆ ---               │
│ str ┆ i64 ┆ i64  ┆ i64 ┆ f64               │
╞═════╪═════╪══════╪═════╪═══════════════════╡
│ foo ┆ 1   ┆ 2    ┆ 1   ┆ 1.0               │
│ bar ┆ 2   ┆ 1    ┆ 2   ┆ 2.0               │
│ foo ┆ 3   ┆ null ┆ 3   ┆ 3.0               │
└─────┴─────┴──────┴─────┴───────────────────┘
    • 2

  2. Há uma questão aberta para solicitar median_horizontal(e outras funções).

    Enquanto isso, uma solução alternativa sugerida é concat_list+ map_elements:

    df.with_columns(
        pl.concat_list(pl.col(pl.Int64))
        .map_elements(lambda l: np.nanmedian(l), return_dtype = pl.Float64)
        .alias("Horizontal Median")
    )

    Alternativamente , com concat_list++ :explodeover

    df.with_columns(
    pl.concat_list(pl.col(pl.Int64))
    .explode().median().over(pl.int_range(pl.len()))
    .alias("Horizontal Median")
    )

    Saída:

    ┌─────┬─────┬──────┬─────┬───────────────────┐
│ ABC ┆ A   ┆ B    ┆ C   ┆ Horizontal Median │
│ --- ┆ --- ┆ ---  ┆ --- ┆ ---               │
│ str ┆ i64 ┆ i64  ┆ i64 ┆ f64               │
╞═════╪═════╪══════╪═════╪═══════════════════╡
│ foo ┆ 1   ┆ 2    ┆ 1   ┆ 1.0               │
│ bar ┆ 2   ┆ 1    ┆ 2   ┆ 2.0               │
│ foo ┆ 3   ┆ null ┆ 3   ┆ 3.0               │
└─────┴─────┴──────┴─────┴───────────────────┘
    • 1

relate perguntas