Извлечение подмножества из Dataframe

Question

Я импортировал Dataframe, используя pandas, затем написал некоторый код (как показано ниже), чтобы создать дополнительный столбец. Мне нужно извлечь подмножество вновь добавленного столбца, а затем создать оператор if, который позволит мне просмотреть последние 2 значения целых чисел и применить следующую логику:

• Если целое число заканчивается на 9, выполнить сложение +2 (например: 9 + 2 = 11)
• Для чего-либо еще добавьте +1 (например, 4 + 1 = 5)

Это код, который я написал для создания дополнительного столбца:

max_items['ind_short'] = ((max_items['Item'] % 1000).astype(int))

Может кто-нибудь помочь?

Answer 1

Предполагая, что под last number within the column вы подразумеваете значение единиц целого числа, вы можете использовать оператор по модулю %. Затем используйте numpy.where для выполнения сопоставления:

df = pd.DataFrame({'A': [6357, 1325, 549, 12312, 19]})

df['B'] = np.where(df['A'] % 10 == 9, 2, 1)

print(df)

       A  B
0   6357  1
1   1325  1
2    549  2
3  12312  1
4     19  2