Создать новый столбец в pandas df и назначить строковые значения условно

Question

Я новичок в pandas, пытаюсь создать новый столбец в Pandas Dataframe и назначить строковое значение на основе функции, но в результате выдается только 1 значение («жилое») для всех 5000 столбцов.Есть идеи, что случилось с моим кодом?Спасибо

def programType(c):
if c['Primary Property Type - Self Selected'] == 'Multifamily Housing' or 'Residence Hall/Dormitory':
    return 'Residential'

elif c['Primary Property Type - Self Selected'] == 'Bank Branch' or 'Hotel' or 'Financial Office' \
or 'Retail Store' or 'Distribution Center' or 'Non-Refrigerated Warehouse' or 'Fitness Center/Health Club/Gym' \
or 'Mixed Use Property' or 'Self-Storage Facility' or 'Wholesale Club/Supercenter' or 'Supermarket/Grocery Store':
    return 'Commercial'  

elif c['Primary Property Type - Self Selected'] == 'Senior Care Community' or 'K-12 School' or 'College/University' \
or 'Worship Facility' or 'Medical Office' or 'Hospital (General Medical & Surgical)':
    return 'Institutional'

elif c['Primary Property Type - Self Selected'] == 'Manufacturing/Industrial Plant': 
    return 'Industrial'

else:
    return 'Other'

Новый столбец называется «Тип программы»

datav3['Program Type'] = datav3.apply(programType, axis=1)

Answer 1

В pandas лучше всего избегать циклов (применяются циклы под капотом), если существуют векторизованные решения, потому что циклы медленные.

Я пытаюсь переписать ваш код - создайте словарь с выводом и списком значений, переключайте ключисо значениями и вызовом map, последний для несоответствующих значений добавить fillna:

d = {'Residential' :['Multifamily Housing', 'Residence Hall/Dormitory'],
     'Commercial' : ['Bank Branch', 'Hotel' , 'Financial Office' , 'Retail Store', 'Distribution Center', 
                   'Non-Refrigerated Warehouse', 'Fitness Center/Health Club/Gym', 'Mixed Use Property',
                   'Self-Storage Facility', 'Wholesale Club/Supercenter', 'Supermarket/Grocery Store'],
     'Institutional':['Senior Care Community', 'K-12 School', 'College/University', 'Worship Facility',
                     'Medical Office', 'Hospital (General Medical & Surgical)'],
     'Industrial':  ['Manufacturing/Industrial Plant'] }

---

d1 = {k: oldk for oldk, oldv in d.items() for k in oldv}
print (d1)

{
    'Multifamily Housing': 'Residential',
    'Residence Hall/Dormitory': 'Residential',
    'Bank Branch': 'Commercial',
    'Hotel': 'Commercial',
    'Financial Office': 'Commercial',
    'Retail Store': 'Commercial',
    'Distribution Center': 'Commercial',
    'Non-Refrigerated Warehouse': 'Commercial',
    'Fitness Center/Health Club/Gym': 'Commercial',
    'Mixed Use Property': 'Commercial',
    'Self-Storage Facility': 'Commercial',
    'Wholesale Club/Supercenter': 'Commercial',
    'Supermarket/Grocery Store': 'Commercial',
    'Senior Care Community': 'Institutional',
    'K-12 School': 'Institutional',
    'College/University': 'Institutional',
    'Worship Facility': 'Institutional',
    'Medical Office': 'Institutional',
    'Hospital (General Medical & Surgical)': 'Institutional',
    'Manufacturing/Industrial Plant': 'Industrial'
}

---

datav3 = pd.DataFrame({'Program':['Medical Office','Hotel',
                                       'Residence Hall/Dormitory',
                                       'Manufacturing/Industrial Plant','House']})
datav3['Program Type'] = datav3['Program'].map(d1).fillna('Other')
print (datav3)
                          Program   Program Type
0                  Medical Office  Institutional
1                           Hotel     Commercial
2        Residence Hall/Dormitory    Residential
3  Manufacturing/Industrial Plant     Industrial
4                           House          Other

Answer 2

Проблема с вашими циклами if.То, как вы сравниваете после or, неверно.

Запись or 'Residence Hall/Dormitory' всегда будет true, следовательно, только первый if оценивается каждый раз, и вы получаете Residential во всех строках.

Вместо этого:

if c['Primary Property Type - Self Selected'] == 'Multifamily Housing' or 'Residence Hall/Dormitory':

Сделайте это:

if c['Primary Property Type - Self Selected'] == 'Multifamily Housing' or c['Primary Property Type - Self Selected'] == 'Residence Hall/Dormitory':

ИЛИ

if any([c['Primary Property Type - Self Selected'] == 'Multifamily Housing', c['Primary Property Type - Self Selected'] == 'Residence Hall/Dormitory']):

Просто внесите вышеуказанное изменение, и ваш коддолжен делать то, что ожидается.Надеюсь, это понятно.