使用数字列作为索引查找字符串的字符

Question

我知道我错过了一些小事所以来这里寻求帮助而不是继续旋转我的轮子 -

我从这里的字符串中找到了我想要的字符的索引：

df['letter_idx']= df['string'].str.find(':')-1 

那么我怎样才能在这个位置找到角色呢？ （我知道下面是不正确的，但是像这样）

df['letter'] = df['string']['letter_idx']

有没有更简单的方法来做到这一点？

Answer 1

您可以将apply方法与“string”、“letter_idx”列一起使用。

df["letter"] = df[["string", "letter_idx"]].apply(lambda row: row.string[row.letter_idx], axis=1)

或者您可以使用zip function。

>>> df = pandas.DataFrame({"string": ["MyString:123", "Another One: 321"]})
>>> df['letter_idx']= df['string'].str.find(':')-1
>>> df["letter"] = [s[i] for s, i in zip(df.string, df.letter_idx)]
>>> df
             string  letter_idx letter
0      MyString:123           7      g
1  Another One: 321          10      e