Получение первых 2 чисел из значения 5 чисел и ввод их в новый столбец в пандах

Question

Я часами искал решение этой проблемы: я хотел бы отсортировать столбец, состоящий из 5 чисел (целых чисел).Затем я хочу использовать первые 2 числа этого значения для группировки.тогда я хочу посчитать эти группировки.

Есть ли простой способ сделать это?Я использую это для подсчета:

print(worksheet['postalcolumn'].value_counts())

Столбец выглядит так: 74 660, 74 5667, 78 320, 71 345 Я хочу новый столбец, как этот 74, 74, 78, 71

Answer 1

Преобразуйте dtype столбца в строку и используйте слайсер str. Вы можете использовать:

worksheet['new_col']=worksheet['postalcolumn'].astype(str).str[:2].astype(int)

Answer 2

Предполагая, что worksheet['postalcolumn'] имеет тип list, вы можете сделать:

worksheet['new_col'] = [] # create an empty new column
for id in worksheet['postalcolumn']:
    first_two = int(str(id)[:2]) # take the first two digits
    worksheet['new_col'].append(first_two) # add the two digits to the new column