主页 / coding / 问题 / 78566724
Accepted
apostofes
Asked: 2024-06-02 22:28:39 +0800 CST

从极坐标列表中选择一组特定的字符串

  • 772
df = pl.DataFrame({'list_column': [['a.xml', 'b.xml', 'c', 'd'], ['e.xml', 'f.xml', 'g', 'h']]})

def func(x):
    return [y for y in x if '.xml' in y]

df.with_columns(pl.col('list_column').map_elements(func, return_dtype=pl.List(pl.String)))

有没有办法在不使用的情况下实现相同的目的map_elements？

python

1 个回答

  1. Best Answer

    有一个已接受的请求list.filter()

    您可以使用以下方式模拟该行为list.eval()

    df.with_columns(
   pl.col('list_column').list.eval(
      pl.element().filter(pl.element().str.ends_with('.xml'))
   )
)

    shape: (2, 1)
┌────────────────────┐
│ list_column        │
│ ---                │
│ list[str]          │
╞════════════════════╡
│ ["a.xml", "b.xml"] │
│ ["e.xml", "f.xml"] │
└────────────────────┘
    • 2

相关问题