使用条件语句删除字符串中的前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]