Python 以特定字符开头的行提取元素

我目前正在处理这个数据帧python:

数据集有一列和n行

我想提取specifics行的specifics组件，例如：

对于以“n”开头的每一行i，将行i的第二个元素存储在变量x中

或

对于以“e”开头的每一行i，将行i的第二个和第三个元素存储在变量x中

我想知道我可以使用哪个函数/操作来解决此问题。

创建简单示例：

d = pd.DataFrame({'a': ['aaaak', 'k jhs', 'anhdga', 'kjdhs']})

您可以使用column.str并看到第一个字母：

data.a.str[0]

out:

0    a
1    k
2    a
3    k

0     True
1    False
2     True
3    False

        a
0   aaaak
2  anhdga

0    a
2    h

您可以检查字母是什么：

data.a.str[0] == 'a'

out:

0    a
1    k
2    a
3    k

0     True
1    False
2     True
3    False

        a
0   aaaak
2  anhdga

0    a
2    h

您可以仅使用第一个字母“a”来调用raws：

data[data.a.str[0] == 'a']

out:

0    a
1    k
2    a
3    k

0     True
1    False
2     True
3    False

        a
0   aaaak
2  anhdga

0    a
2    h

然后你可以得到另一个以字母“a”开头的字母：

data[data.a.str[0] == 'a'].a.str[2]

out:

0    a
1    k
2    a
3    k

0     True
1    False
2     True
3    False

        a
0   aaaak
2  anhdga

0    a
2    h

---

请提供您所了解的内容和样本输出？