Я хочу создать новые столбцы во фрейме данных, содержащие данные о велосипедных поездках (названия начальной и конечной станций, а также продолжительность поездки). Я хочу назначить разные станции определенному району. Однако мне не удается применить эту функцию как к столбцам начальной, так и к конечной станции. Я был бы признателен за любые рекомендации. Ниже мой код:
Код: Выделить всё
#Assign stations to neighborhood
Shadyside = ['Ivy St & Walnut St', 'Shady Ave & Ellsworth Ave','S Negley Ave & Centre Ave','Penn Ave & S Whitfield St',
'Pierce St & Summerlea St']
#print(Shadyside)
def station(x):
for i in Shadyside:
#print(i)
if re.search(r'i', x) != None:
return 'Yes'
else:
return 'No'
Biking_data['Shadyside'] = Biking_data['Start Station'].apply(station)
print(Biking_data)
Я попробовал приведенный выше код только для одного столбца, чтобы проверить, работает ли функция. Но тогда говорят, что станции, расположенные не в Шейдисайде, находятся по соседству.
Подробнее здесь:
https://stackoverflow.com/questions/790 ... data-frame