Pandas сделать новый столбец из строкового среза другого столбца

Я хочу создать новый столбец в Pandas, используя строку, разрезанную для другого столбца в фрейме данных.

Например.

Sample  Value  New_sample
AAB     23     A
BAB     25     B

Где New_sample - новый столбец, сформированный из простого фрагмента [:1] Sample

Я пробовал несколько вещей безрезультатно - я чувствую, что мне не хватает чего-то простого.

Какой самый эффективный способ сделать это?

Ответы

Ответ 1

Вы можете вызвать метод str и применить срез, это будет намного быстрее, чем другой метод, поскольку это векторизован (спасибо @unutbu):

df['New_Sample'] = df.Sample.str[:1]

Вы также можете вызвать лямбда-функцию на df, но это будет медленнее на больших кадрах данных:

In [187]:

df['New_Sample'] = df.Sample.apply(lambda x: x[:1])
df
Out[187]:
  Sample  Value New_Sample
0    AAB     23          A
1    BAB     25          B