Python 在列中查找列名

即使在列重复的情况下，此操作也有效

df = pl.DataFrame({
    'a': [1, 2, 1,3],
    'b': [2, 1, 2,4],
    'c': [3, 3, 2,5],
    'column': ['a', 'c', 'b','a'],
})
df.with_row_count('i').join(
    df.with_row_count('i')
        .melt(['i','column'], value_name='lookup')
        .filter(pl.col('column')==pl.col('variable'))
        .select('i','lookup'),
    on='i', how='left').sort('i').drop('i')

shape: (4, 5)
┌─────┬─────┬─────┬────────┬────────┐
│ a   ┆ b   ┆ c   ┆ column ┆ lookup │
│ --- ┆ --- ┆ --- ┆ ---    ┆ ---    │
│ i64 ┆ i64 ┆ i64 ┆ str    ┆ i64    │
╞═════╪═════╪═════╪════════╪════════╡
│ 1   ┆ 2   ┆ 3   ┆ a      ┆ 1      │
│ 2   ┆ 1   ┆ 3   ┆ c      ┆ 3      │
│ 1   ┆ 2   ┆ 2   ┆ b      ┆ 2      │
│ 3   ┆ 4   ┆ 5   ┆ a      ┆ 3      │
└─────┴─────┴─────┴────────┴────────┘

这里有另一种不使用melt的方法

df.join(
    df.select(pl.col('column').unique().sort()).with_row_count('i'), 
    on='column'
).with_columns(
    lookup=(pl.concat_list('a','b','c').list.get(pl.col('i')))
).drop('i')

它的工作原理是将‘a’、‘b’、‘c’列转换为列表列，然后使用.get来提取该列表的索引，因为GET将接受一个表达式.要做到这一点，我们首先必须创建一个索引值ABC是什么的查找表.如果现有列与column中的列不完全匹配，它显然会严重崩溃.假设你的数据被提前擦除，那么我认为这会比融化更快.