Sou novo em Polars e Python em geral. Tenho um problema um tanto incomum para o qual preciso de ajuda. Eu tenho um dataframe com mais de 50 colunas que são 0/1. Preciso criar uma nova coluna que contenha uma lista separada por vírgulas de cada coluna que contenha 1, mas usando parte do nome da coluna. Se hccx = 1, anexe x a uma coluna de string. Um exemplo simplificado:
df=pl.DataFrame(
{'id':[1,2,3], 'hcc1':[0,1,1],'hcc2':[0,0,1],'hcc5':[0,1,1],'hcc8':[1,0,0]}
)
shape: (3, 5)
┌─────┬──────┬──────┬──────┬──────┐
│ id ┆ hcc1 ┆ hcc2 ┆ hcc5 ┆ hcc8 │
│ --- ┆ --- ┆ --- ┆ --- ┆ --- │
│ i64 ┆ i64 ┆ i64 ┆ i64 ┆ i64 │
╞═════╪══════╪══════╪══════╪══════╡
│ 1 ┆ 0 ┆ 0 ┆ 0 ┆ 1 │
│ 2 ┆ 1 ┆ 0 ┆ 1 ┆ 0 │
│ 3 ┆ 1 ┆ 1 ┆ 1 ┆ 0 │
└─────┴──────┴──────┴──────┴──────┘
Quero criar uma nova coluna (tipo string), hccall, semelhante a esta:
| eu ia | ligar |
|---|---|
| 1 | 8 |
| 2 | 1,5 |
| 3 | 1,2,5 |
Imagino que algum tipo de compreensão de lista percorrendo colunas que começam com 'hcc' funcionaria, mas estou meio preso. Posso criar um loop, mas não sei como anexar à coluna dentro do loop. Alguma ideia inteligente?