使用条件语句删除字符串中的前3个字符

Question

任何人都可以帮忙吗？ 我正在尝试使用以下语句删除字符串中的前三个字符：

Data['COUNTRY_CODE'] = Data['COUNTRY1'].str[3:]

这将在删除字符串的前三个值后创建一个新列。但是，我不希望将此方法应用于同一列中的所有值，因此希望有一种方法可以使用条件语句（例如“ Where”）来仅更改所需的字符串？

Answer 1

我假设您正在使用熊猫，所以您的状况检查可以像这样：

condition_mask = Data['COL_YOU_WANT_TO_CHECK'] == 'SOME CONDITION'

您的新列可以创建为：

# Assuming you want the first 3 chars as COUNTRY_CODE
Data.loc[condition_mask, 'COUNTRY_CODE'] = Data['COUNTRY1'].str[:3]