I have a column "filename" where there are values named "train_XXXXX.jpg" / "test_XXXX.jpg". I want to create a new column "