從特定字符開始逐行提取元素

Question

我目前正在研究這個 DataFrame python ：

數據集有 1 列和 n 行。

我想提取特定行的特定組件，例如：

對於以“n”開頭的每一行 i，將行 i 的第二個元素存儲在變量 x 中。

或者

對於以“e”開頭的每一行 i，將行 i 的第二個和第三個元素存儲在變量 x 中。

我想知道我可以使用哪個函數/操作來解決這個問題。

Answer 1

創建簡單示例：

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

您可以使用 column.str 並查看第一個字母：

data.a.str[0]

出去：

0    a
1    k
2    a
3    k

你可以檢查這封信是什么：

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

出去：

0     True
1    False
2     True
3    False

你可以只用第一個字母 'a' 調用 raws：

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

出去：

        a
0   aaaak
2  anhdga

然后你可以得到另一個從 'a' 開始的 raws 字母：

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

出去：

0    a
2    h