如何编写基于子字符串选择单元格的列表理解[重复]

【问题标题】：how to write list comprehension for selecting cells base on a substring [duplicate]如何编写基于子字符串选择单元格的列表理解[重复]
【发布时间】：2019-09-04 19:31:53
【问题描述】：

我正在尝试使用列表理解在一行中重写以下内容。我只想选择包含子字符串 '[edit]' 的单元格。 ut 是我的数据框，我要从中选择的列是“col1”。谢谢！

    for u in ut['col1']:
        if '[edit]' in u:
            print(u)

I expect the following output:
Alabama[edit]
Alaska[edit]
Arizona[edit]
...

【问题讨论】：

见this answer，向下滚动到“一个很好的选择：使用列表理解！”
@ChrisA 我试过了，但出现语法错误。我认为这与 'u in ut['col1'] 与列表理解的排序有关

标签： pandas substring list-comprehension

【解决方案1】：

如果 Pandas Series 的输出是可以接受的，那么您可以直接使用 .str.contains，而无需循环

s = ut[ut["col1"].str.contains("edit")]

如果您需要单独打印Series 的每个元素，则循环遍历Series using

for i in s:
    print(i)

【讨论】：