python - 如何获取列值以任何给定字符开头的 Pandas Dataframe 行

Question

我有以下 Pandas 数据框:

d = {'col1': ["aaa", "bbb", "ccc", "ddd", "acc", "bcc"]}
df = pd.DataFrame(data=d)
df

以上代码的输出:

    col1
0   aaa
1   bbb
2   ccc
3   ddd
4   abb
5   bcc

我需要获取列值以“a”或“c”开头的行。过滤后的结果应该如下所示:

    col1
0   aaa
1   ccc
2   abb

如何在不使用 for 循环的情况下实现这一点？

Answer 1

仅适用于a 和c

df = df[df["col1"].str.match('^[ac]')]

您可以在方括号中包含您需要匹配的任意多个字母。

df = df[df["col1"].str.match('^[abcdefg]')]