如何在lambda python中使用for循环和if？

【问题标题】：How to use for loop along with if inside lambda python?如何在lambda python中使用for循环和if？
【发布时间】：2021-07-01 02:00:59
【问题描述】：

我有一个数据框 df 有一列 tags 。 tags 列的每个元素都是一个字典列表，如下所示：

[
    {
        "id": "leena123",
        "name": "LeenaShaw",
        "slug": null,
        "type": "UserTag",
        "endIndex": 0,
        "startIndex": 0
    },
    {
        "id": "1234",
        "name": "abc ltd.",
        "slug": "5678",
        "type": "StockTag",
        "endIndex": 0,
        "startIndex": 0
    }
]

列表可以包含任意数量的元素。

样本数据集：

 0  some_data  [{'id': 'leena123', 'name': 'leenaShaw', 'slug': None, 'type...
 1  some data  [{'id': '6', 'name': 'new', 'slug': None, 'type...

我想从 type 为 UserTag 列的 tags 列中创建所有 id 的列表>

样本输出： ['leena123', 'saily639,...]

我正在尝试这个： list(df['tags'].apply(lambda x: d['name'] if any(d['type'] == 'UserTag' for d in x)))

但它不起作用。请帮帮忙。

【问题讨论】：

请分享您的数据框样本和预期输出。

标签： python pandas dataframe dictionary lambda

【解决方案1】：

将List Comprehension 与df.apply 一起使用：

df['id'] = df.tags.apply(lambda x: [i['id'] for i in x if i.get('type') == 'UserTag'])

从id 列创建list：

import itertools

l = df['id'].values.tolist()
output_id_list = list(itertools.chain(*l))

如果您想从df 中删除id 列，请执行以下操作：

df.drop('id', inplace=True)

【讨论】：

这是用用户标签创建一个新列。我想要一个用户标签列表如下['leena123', 'saily639,...]
您要替换现有的tags 列吗？
没有。我不添加新列或替换现有标签。我想要一个id 的列表，其中type 是userTag。
@dipanjana 请检查我更新的答案。