Сравнение столбцов DataFrame с другим списком и изменение

Question

Good Day,

У меня есть Pandas DataFrame со столбцом «Cityname» и другим списком «BigCities». Для каждой строки в столбце «Cityname» я хотел бы поменять имя города, когда city ​​находится в списке 'BigCities' на 1 или, если нет, 0.

Может кто-нибудь помочь с этим. У меня проблемы при сравнении двух списков.

Спасибо.

С уважением

Питер

Answer 1

Вы можете проверить членство по списку BigCities с помощью Series.isin, а затем привести к целым числам:

df['Cityname'] = df['Cityname'].isin(BigCities).astype(int)

Или использовать numpy.where:

df['Cityname'] = np.where(df['Cityname'].isin(BigCities), 1, 0)