Как создать новый столбец с помощью других столбцов с априорными ограничениями на содержимое?

Question

[Timebird]

Имеется датафрейм, в первом столбце данные могут быть и int, и str.

Col1       Col2       Col3
автобус      4       100500
37           2       100500
57           3       100500
троллейбус   4       100500

Требуется: создать новый столбец, чтобы он был как первый, но только если данные соответствующей строки первого столбца - типа str

Col1       Col2       Col3        Col4
автобус      4       100500       автобус
37           2       100500       NaN
57           3       100500       NaN
троллейбус   4       100500       троллейбус

Как это сделать?

Answer 1

[Дмитрий]

не numpy, конечно, но с обычным двумерным массивом я бы сделал так:

base = [
    ['автобус', 4, 100500],
    [37, 2, 100500],
    [57, 3, 100500],
    ['троллейбус', 4, 100500]
    ]

for i in base:
    i.append(i[0] if isinstance(i[0], str) else 'NaN')