如何根据子串[duplicate]选择单元格来编写列表理解

Question

这个问题在这里已有答案：

• Select by partial string from a pandas DataFrame 8个答案

我试图使用列表理解在一行中重写以下内容。我想选择仅包含子串'[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]
...

Answer 1

如果Pandas Series的输出是可以接受的，那么你可以使用.str.contains，没有循环

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

如果你需要分别打印Series的每个元素，然后循环Series using

for i in s:
    print(i)