Как добавить строку в середине столбца в пандах

Question

Я ищу решение, которое не включает в себя функцию .apply или лямбду, которая просматривает список и рекламирует строку с нужным индексом.У меня есть такой столбец со многими записями:

df = pd.DataFrame(["1:77631829:-:1:77641672:-"], columns=["position"])

    position
0   1:77631829:-:1:77641672:-

Я бы хотел:

    position
0   chr1:77631829:-:chr1:77641672:-

Поэтому вставьте «chr» в начале и после третьего двоеточия :

Я бы подумал, что что-то подобное будет, но вставка не была реализована последовательно:

"chr" + df["position"].str.split(":").insert(3, "chr").str.join(":")

Это делает это, но выглядит неэффективно:

"chr" + df["position"].str.split(":").str[:3].str.join(":") + "chr" + df["position"].str.split(":").str[3:].str.join(":")

Answer 1

Я думаю, что вы можете использовать разделение на 3 значение :, затем извлечь заголовок и хвост списков - соединить заголовок, добавить ch к хвосту, добавить ch и последний добавить к списку L:

df = pd.DataFrame(["1:77631829:-:1:77641672:-","1:77631829:-:1:77641672:-"], 
                  columns=["position"])
print (df)
                    position
0  1:77631829:-:1:77641672:-
1  1:77631829:-:1:77641672:-

L = []
for x in df["position"]:
    *i, j = x.split(':', 3)
    L.append(("chr" + ':'.join(i) + "chr" + j))

df['new'] = L
print (df)
                    position                             new
0  1:77631829:-:1:77641672:-  chr1:77631829:-chr1:77641672:-
1  1:77631829:-:1:77641672:-  chr1:77631829:-chr1:77641672:-

Взломать решение с комментариями:

'chr' + df['position'].str.replace('-:', '-:chr')

Быстрее с пониманием списка и f-строк:

df['new'] = [f"ch{x.replace('-:', '-:chr')}" for x in df['position']]

Производительность :

df = pd.DataFrame(["1:77631829:-:1:77641672:-","1:77631829:-:1:77641672:-"], 
                  columns=["position"])

#[20000 rows x 1 columns]
df = pd.concat([df] * 10000, ignore_index=True)

In [226]: %%timeit
     ...: L = []
     ...: for x in df["position"]:
     ...:     *i, j = x.split(':', 3)
     ...:     L.append(("chr" + ':'.join(i) + "chr" + j))
     ...: 
     ...: df['new1'] = L
     ...: 
18.9 ms ± 1.25 ms per loop (mean ± std. dev. of 7 runs, 100 loops each)

In [227]: %%timeit
     ...: df['new2'] = "chr" + df["position"].str.split(":").str[:3].str.join(":") + "chr" + df["position"].str.split(":").str[3:].str.join(":")
     ...: 
50.8 ms ± 1.2 ms per loop (mean ± std. dev. of 7 runs, 10 loops each)

In [228]: %%timeit
     ...: df['new3'] = 'chr' + df['position'].str.replace('-:', '-:chr')
     ...: 
21.5 ms ± 140 µs per loop (mean ± std. dev. of 7 runs, 10 loops each)

In [229]: %%timeit
     ...: df['new4'] = [f"ch{x.replace('-:', '-:chr')}" for x in df['position']]
     ...: 
8.59 ms ± 130 µs per loop (mean ± std. dev. of 7 runs, 100 loops each)