drop df columns in a certain list code example
Example: python - dataframe columns is a list - drop
# Df with a coulmn (dims) that contain list
key1 = 'site channel fiscal_week'.split()
key2 = 'site dude fiscal_week'.split()
key3 = 'site eng fiscal_week'.split()
keys = pd.DataFrame({'key': [1,2,3],
'dims': [key1,key2,key3]})
# Output
dims key
[site, channel, fiscal_week] 1
[site, dude, fiscal_week] 2
[site, eng, fiscal_week] 3
# Solution
keys['reduced_dims'] = keys['dims'].apply(
lambda row: [val for val in row if val != 'fiscal_week']
)