Python Pandas: генерировать фиктивную переменную из числовой переменной в соответствии с порогом

Question

Цель состоит в том, чтобы создать новый столбец из df с 1, если значение из столбца "% Renewable" равно или выше медианы, и 0, если значение ниже медианы.

df = pd.DataFrame({'% Renewable': [np.NaN, 12, np.NaN, 11, 17, 62, 18, 15, np.NaN, 2, np.NaN, np.NaN, 6, np.NaN, 70]},
index=['China', 'United States', 'Japan', 'United Kingdom', 'Russian Federation', 'Canada', 'Germany', 'India', 'France', 'South Korea', 'Italy', 'Spain', 'Iran', 'Australia', 'Brazil'])

Я получил медиану:

median = df['% Renewable'].median()

Но что теперь? Должен ли я использовать функцию get_dummies? Или вырезать что ли?

Answer 1

Это должно сделать трюк:

df['new_column'] = df['% Renewable'] >= median