如何在 pandas dataframe 中制作切片？

Question

我想使用列中的字符串“PP”对 dataframe 进行切片，并仅获取字符串之后的数字：

Dataframe：

data = {'Serie':['28PP3097', '23228PP3097', '1822343218PP3097', '43642183097'],
    'FooBar':["foo", "bar", "foo", "bar"]}
df = pd.DataFrame(data)

预期结果：

我尝试：

df["Serie"] = np.where(df["Serie"].str.contains("PP"), df["Serie"][df["Serie"].str.find('PP')+1:],df["Serie"])

在这个 df 中，但它给了我一个错误“无法使用这些索引器对 RangeIndex 进行切片索引”

Answer 1

您可以通过拆分并获取“PP”之后的最后一项来做到这一点：

data = {'Serie':['28PP3097', '23228PP3097', '1822343218PP3097', '43642183097'],
        'FooBar':["foo", "bar", "foo", "bar"]}
df = pd.DataFrame(data)

df['Serie']=[i.split('PP')[-1] for i in df['Serie']]

结果

         Serie FooBar
0         3097    foo
1         3097    bar
2         3097    foo
3  43642183097    bar