Панда добавив несколько дополнительных столбцов на основе многих условий

Question

Я чтение данных в кусках из CSV, как это:

 for chunk in pd.read_csv(file, chunksize=50000, names = col_names, header = 0, dtype = dtype): 
      chunk['derived_field_1'] = [1 if x == 'High' else -1 for x in chunk['indicator']] 

выше работает и она основана на одном условии. Я хочу сделать это на основе условий на двух полях. Создание общего количества комбинаций значений будет 8. В качестве примера

chunk['derived_field_2'] = [chunk['column_1'] if ((x == 'Red' for x in chunk['Color']) and (y == 'High' for y in chunk['Indicator'])) else 
          chunk['column_2'] if ((x == 'Green' for x in chunk['Color']) and (y == 'Low' for y in chunk['Indicator'])) else 0] 

Я хочу сделать выше и продолжать идти с условием еще для более 6 условий, как выше. Это не работает, два цикла не работают. Я получаю эту ошибку -

raise ValueError('Length of values does not match length of ' 'index') 
ValueError: Length of values does not match length of index 

Может кто-нибудь знать причину этой ошибки?

Answer 1

Вы можете использовать numpy.where для векторизованного решения:

import numpy as np 
chunk['derived_field_2'] = (np.where((chunk['Color'] == "Red") & (chunk["Indicator"] == "High"), chunk["column_1"], 
    np.where((chunk['Color'] == "Green") & (chunk["Indicator"] == "Low"), chunk["column_2"], 0))