Python Объединение двух столбцов в один на основе значения каждого столбца

Question

Я работаю с этим набором данных, и я хотел бы объединить yr_built и yr_renovated в один, предпочтительно в yr_built, исходя из этого: если значение в yr_renovated больше 0, то я бы хотел иметь это значение, иначе значение yr_built.

Не могли бы вы помочь мне в этом?

Спасибо!

Answer 1

Вот, пожалуйста. В основном вам нужны pandas для фрейма данных, затем создайте новый столбец, используя numpy, чтобы проверить, является ли значение 'yr_renovated' больше нуля, иначе используйте 'yr_built'

import pandas as pd
import numpy as np
df = pd.read_csv('https://raw.githubusercontent.com/Jonasyao/Machine-Learning-Specialization-University-of-Washington-/master/Regression/Assignment_four/kc_house_data.csv', error_bad_lines=False)
df=df[['yr_built','yr_renovated','date','bedrooms']]
newdata['MyYear']=np.where(df['yr_renovated'] > 0,df['yr_renovated'],df['yr_built'])
newdata